yukicoder : No.323 yuki国

今年は今のところ暖冬気味ですね…。
http://yukicoder.me/problems/845

問題

H*Wの2次元グリッドが与えられる。
各セルについて、雪が積もっているかどうか入力が与えられる。

初期状態で(Si,Sj)のマスに大きさAの雪玉がある。
この雪玉を隣接マスを辿り転がしていく。
その際、雪があるマスに到達するたびに大きさが+1、雪のないマスに到達するたびに大きさが-1する。
雪玉の大きさが途中0にならないようにして、大きさBの雪玉が(Gi,Gj)に来るようにできるか判定せよ。

解法

現在位置と雪玉のサイズを状態とし、DFSなりBFSすれば良い。
雪玉のサイズがH*Wを超えても意味がないので、大きくなる過ぎる場合は探索を打ち切ろう。

int H,W;
int A,SY,SX;
int B,GY,GX;
string M[51];

int ok[51][51][3600];


void solve() {
	int i,j,k,l,r,x,y; string s;
	
	cin>>H>>W;
	cin>>A>>SY>>SX;
	cin>>B>>GY>>GX;
	FOR(y,H) cin>>M[y];
	
	
	ok[SY][SX][A]=1;
	queue<int> Q;
	Q.push(A*10000+SY*100+SX);
	
	while(Q.size()) {
		int r=Q.front();
		Q.pop();
		int sz=r/10000;
		int cy=r/100%100;
		int cx=r%100;
		
		FOR(i,4) {
			int dd[]={0,1,0,-1};
			int ty=cy+dd[i];
			int tx=cx+dd[i^1];
			if(ty<0 || ty>=H || tx<0 || tx>=W) continue;
			int nz=sz+((M[ty][tx]=='*')?1:-1);
			if(nz<=0 || nz>3500) continue;
			if(ok[ty][tx][nz]) continue;
			ok[ty][tx][nz]=1;
			Q.push(nz*10000+ty*100+tx);
		}
	}
	
	if(ok[GY][GX][B]) _P("Yes\n");
	else _P("No\n");
	
}

まとめ

雪玉？雪球？