#120597 (C++11) No.424 立体迷路

• 問題: No.424 立体迷路
• ユーザー: HAHAHA
• 提出日時: 2016-10-01 23:23:16
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 1,346 bytes
• コンパイル時間: 1,343 ms
• コンパイル使用メモリ: 165,356 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-05 07:11:04
• 合計ジャッジ時間: 2,142 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	1 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	2 ms; 5,376 KB
testcase_21	AC	2 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	2 ms; 5,376 KB
testcase_25	AC	2 ms; 5,376 KB

ソースコード

#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;
}