夢追い人

"It takes a dreamer to make a dream come true."―Vincent Willem van Gogh

パティシエ

読み間違えてたorz

AOJ 0120

解説

グラフに落としこむのどうやんだとおもったら小さいものがハマるケースがないようで・・・
この条件で一気に簡単になりまふ
具体的には各円が隣接したときそれぞれの半径をr1,r2と置くとそれぞれの中心間の横幅(直接の距離ではなく)が√((r1+r2)^2-(r1-r2)^2) で求まるので、各円同士でこの距離をまず計算しておきます。注意しておきたいのは、壁の処理でこれは番兵的な感じで半径0の円として持っておきます。つまり壁と各円の中心が要素、それぞれの横の座標においての距離が辺の重みになるわけです。
それを求め終わったらあとはTSPそのものです。最短の距離を計算し、それと制限幅の大小比較で判定します。

コード

入力が鬼畜

#include <cstdio>
#include <cmath>
#include <cstring>
#include <climits>
#include <iostream>
#include <sstream>
#include <string>
#include <algorithm>
using namespace std;
#define inf INT_MAX/2
int n, haba;
string in;
int r[13];
double dist[13][13], dp[1<<13][13];
void input(string in) {
    int i;
    string x="";
    for (i=0; i<in.length(); i++) {
        if (in[i]==' ') {
            stringstream ss;
            ss<<x;
            ss>>haba;
            x="";
            break;
        } else {
            x+=in[i];
        }
    }
    for (;i<in.length(); i++) {
        if (in[i]==' ') {
            stringstream ss;
            ss<<x;
            ss>>r[n];
            x=""; n++;
        } else {
            x+=in[i];
        }
    }
    stringstream ss;
    ss<<x;
    ss>>r[n]; n++;
}
double calc(int i, int j) {
    return sqrt((double)(r[i]+r[j])*(r[i]+r[j])-(double)(r[i]-r[j])*(r[i]-r[j]));
}
int main() {
    while (getline(cin,in)) {
        n=0; memset(r,0,sizeof(r));
        input(in);
        for (int i=1; i<n; i++) {
            dist[0][i]=dist[i][0]=r[i];
            for (int j=i+1; j<=n; j++) {
                dist[i][j]=dist[j][i]=calc(i,j);
            }
        }
        for (int S=0; S< 1<<n; S++) {
            fill(dp[S], dp[S]+n, inf);
        }
        dp[(1<<n)-1][0]=0;
        for (int S=(1<<n)-2; S>=0; S--) {
            for (int v=0; v<n; v++) {
                for (int u=0; u<n; u++) {
                    if (!(S>>u&1)) {
                        dp[S][v]=min(dp[S][v],dp[S|1<<u][u]+dist[v][u]);
                    }
                }
            }
        }
        if (dp[0][0]>haba) puts("NA");
        else puts("OK");
    }
}
広告を非表示にする