結果

問題 No.424 立体迷路
ユーザー HAHAHA
提出日時 2016-10-01 15:02:50
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,344 bytes
コンパイル時間 1,724 ms
コンパイル使用メモリ 175,248 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-21 12:26:51
合計ジャッジ時間 2,572 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 5
other WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;


struct UnionFind{
	vector<int> data;
		void init(int n){ data.assign(n,-1); }
		bool unionSet(int x, int y){
			x = root(x); y = root(y);
			if(x!=y){
				if(data[y] < data[x]) swap(x,y);
				data[x] += data[y]; data[y] = x;
			}
			return x != y;
		}
		bool findSet(int x, int y){ return root(x) == root(y); }
		int root(int x){ return data[x] < 0 ? x : data[x] = root(data[x]); }
		int size(int x){ return -data[root(x)]; }
};


int main(){
	int h,w;
	cin >> h >> w;
	int sy,sx,gy,gx;
	cin >> sy >> sx >> gy >> gx;
	sy--,sx--,gy--,gx--;
	vector<string> b(h);
	for(int i=0;i<h;i++){
		string s;
		cin >> s;
		b[i]=s;
	}
	
	UnionFind uf; uf.init(h * w);
	for(int i=0;i<h;i++){
		for(int j=0;j<w;j++){
			static const int dy[]={0, 1, 0, -1}, dx[]={1, 0, -1, 0};
			for(int d=0;d<4;d++){
				int yy = i+dy[d], xx = j+dx[d];
				if(yy < 0 || yy >= h || xx < 0 || xx >=w) continue;
				if(abs(b[yy][xx] - b[i][j]) <=1){
					uf.unionSet(i*w+j, yy*w+xx);				
				}
				else if(b[yy][xx] < b[i][j]){
					int yyy = yy + dy[d], xxx = xx + dx[d];
					if(yyy < 0 || yyy >=h || xxx < 0 || xxx >= w) continue;
					if(b[yyy][xxx] == b[i][j])
					uf.unionSet(i * w + j, yyy * w + xxx);
				}
			}
		}
		bool ans = uf.findSet(sy * w + sx, gy * w + gx);
		cout << (ans ? "YES" : "NO") << endl;
	}
	return 0;
}
0