1250. 格子游戏 (降维并查集)

格子游戏

裸并查集
二维坐标一维化 ：x*n + y （x，y均从0开始，即从0开始给每个点编号）
当开始的点与下一个点在同一个集合内，就是已经形成环。

#include<iostream>
using namespace std;

int p[40010];
int n,m;
int find(int x){
	if(p[x] !=x) p[x] = find(p[x]);
	return p[x];
}

int main()
{
	cin>>n>>m;
	for(int i=0;i<n*n;i++) p[i] = i;
	int num=0,ans=0;
	while(m--){
		int x,y;
		char c;
		cin>>x>>y>>c;
		x--,y--;
		int a = x*n + y , b;
		if(c=='D') b = (x+1)*n+y;
		else b = x*n+y+1;
		int pa = find(a),pb = find(b);
		num++;
		if(pa==pb&&!ans){
			ans = num;	
		}
		else p[pa] = pb;
	}
	if(!ans) puts("draw");
	else cout<<ans<<endl;
	return 0;
}