問題
http://community.topcoder.com/stat?c=problem_statement&pm=12548&rd=15656
・友達が複数存在し、互いに誰と誰が友達であるかがわかっている。
・パーティにて、ダンスが任意の回数行われる。ダンスは2人ペアで行われる。
・ペアとなれるのは友達同士か、もしくは共通の友達がいる必要がある。
・一度ダンスを行えば、そのペアは友達となる。
・このとき、0番目の人と1番目の人が友達になるまでに必要な
最小のダンス回数を求める。
解き方
・最短経路を求める問題なので、ワーシャルフロイドが使えそう。
・サンプルを見ると、0と1との最短経路-1が答え?
・サンプルは通ったので提出。
→SystemFailed.
・問題文を良く見ると、毎回のダンスで複数の友達がペアになるので、
初期状態の距離だけでは答えは求まらない。
・具体的には「0-A-B-C-D-1」のときは1回のダンスで「0-B-C-1」となる。
・つまり、最初の距離dから(d+1)/3ずつ引いていき、1以下になるまでのダンス回数が
答えになる。
コード
using namespace std;
#define all(c) (c).begin(),(c).end()
#define FORE(i,d,e) for(int i=d;i<e;i++)
#define FOR(i,s,e) for (int i = int(s); i != int(e); i++)
#define FORIT(i,c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); i++)
#define ISEQ(c) (c).begin(), (c).end()
class DancingFoxes {
public: int minimalDays(vector<string> friendship) {
int n=friendship.size();
int d[n][n];
FORE(i,0,n)FORE(j,0,n)d[i][j]=1e+9;
FORE(i,0,n)FORE(j,0,n)if(friendship[i][j]=='Y')d[i][j]=1;
FORE(k,0,n)FORE(i,0,n)FORE(j,0,n)d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
if(d[0][1]==1e+9)return -1;
int p=d[0][1],ret=0;
while(p>1){
p-=(p+1)/3;
ret++;
}
return ret;
}
};
#define all(c) (c).begin(),(c).end()
#define FORE(i,d,e) for(int i=d;i<e;i++)
#define FOR(i,s,e) for (int i = int(s); i != int(e); i++)
#define FORIT(i,c) for (typeof((c).begin()) i = (c).begin(); i != (c).end(); i++)
#define ISEQ(c) (c).begin(), (c).end()
class DancingFoxes {
public: int minimalDays(vector<string> friendship) {
int n=friendship.size();
int d[n][n];
FORE(i,0,n)FORE(j,0,n)d[i][j]=1e+9;
FORE(i,0,n)FORE(j,0,n)if(friendship[i][j]=='Y')d[i][j]=1;
FORE(k,0,n)FORE(i,0,n)FORE(j,0,n)d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
if(d[0][1]==1e+9)return -1;
int p=d[0][1],ret=0;
while(p>1){
p-=(p+1)/3;
ret++;
}
return ret;
}
};