問題文がちょっとややこしいな。
https://yukicoder.me/problems/no/1400
問題
N頂点の無向グラフが与えられる。
任意の頂点から、ちょうどD回辺に沿って移動したとき、任意の点で止まることができるか。
解法
無向グラフなので、同じ頂点間を往復すれば移動回数を2回ずつ消費できる。
そこで、
f(x,y,o) := 頂点xから頂点yに、移動回数の偶奇がo(偶数または奇数)を満たす最小の移動回数
を求めよう。すべての頂点対(x,y)についてf(x,y,D%2)≦Dが満たせるかを判定すればよい。
あとはf(x,y,o)を埋めるわけだが、これは普通に解くとO(N^3)かかる。
Bitsetを使い高速化しよう。bitset中で位置を求めるのに_Find_first()を使うとよい。
int N; ll T; string S; bitset<2040> E[2020]; int D[2020][2020][2]; void solve() { int i,j,k,l,r,x,y; string s; cin>>N>>T; FOR(y,N) { cin>>S; FOR(x,N) { E[y][x]=S[x]=='1'; D[y][x][0]=D[y][x][1]=101010; } } FOR(i,N) { bitset<2040> B[2],nex[2]; nex[0][i]=1; FOR(j,2*N+1) { bitset<2040> diff=B[j&1]^nex[j&1]; while(diff._Find_first()<N) { x=diff._Find_first(); diff[x]=diff[x]^1; B[j&1][x]=1; D[i][x][j&1]=j; nex[(j&1)^1] |= E[x]; } } } FOR(y,N) FOR(x,N) { if(D[y][x][T%2]>T) return _P("No\n"); if(D[y][x][T%2]==101010) return _P("No\n"); } cout<<"Yes"<<endl; }
まとめ
なんで_Find_firstはアンダースコアで始まってたり大文字含んでたりするんだろ。
非標準な関数なのかな。