【并查集】格子游戏

原题链接

并查集裸题。如果将要用线连接的两个点位于同一个集合内，那么当前操作可以围成一个封闭的圈。否则，将两点合并到同一集合内。

代码实现：

#include<bits/stdc++.h>
#define N 1000000+10
using namespace std;

int n,m,ans=0;
int f[N];

int get(int x){
    if(f[x]==x)return x;
    return f[x]=get(f[x]);
}

void merge(int a,int b){
    f[get(a)]=get(b);
}

int main(){
    cin>>n>>m;
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            f[i*n+j]=i*n+j;
            
    for(int i=1;i<=m;i++){
        int x,y;char ch;
        cin>>x>>y>>ch;
        
        if(ch=='D'){
            if(get(x*n+y)==get((x+1)*n+y)){ans=i;break;}
            merge(x*n+y,(x+1)*n+y);
        }
        
        if(ch=='R'){
            if(get(x*n+y)==get(x*n+y+1)){ans=i;break;}
            merge(x*n+y,x*n+y+1);
        }
    }
    
    if(ans==0)cout<<"draw";
    else cout<<ans;
    return 0;
}