kmjp's blog

競技プログラミング参加記です

yukicoder : No.1400 すごろくで世界旅行

問題文がちょっとややこしいな。
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はアンダースコアで始まってたり大文字含んでたりするんだろ。
非標準な関数なのかな。