結果

問題 No.424 立体迷路
コンテスト
ユーザー HAHAHA
提出日時 2016-10-01 23:23:16
言語 C++11（廃止可能性あり）
(gcc 13.3.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 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