并查集——格子游戏

#include<bits/stdc++.h>
using namespace std;
const int maxn=40010;
int s[maxn];
int n,m;
void init_set(){
    for(int i=0;i<=maxn;i++)
        s[i]=i;
}
int get(int x,int y){
    return x*n+y;
}
int find_set(int x){
    if(s[x]!=x)s[x]=find_set(s[x]);
    return s[x];
}
bool in_same_set(int x,int y){
    x=find_set(x);
    y=find_set(y);
    if(x!=y)
    {
        s[x]=y;
        return false;
    }
    return true;
}
int main(){
    init_set();
    int ans=0;
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int x,y;
        char c;
        cin>>x>>y>>c;
        x--;y--;
        int pa=get(x,y),pb;
        if(c=='D')pb=get(x+1,y);
        else pb=get(x,y+1);
        if(in_same_set(pa,pb)){
            ans=i;
            break;
        }
    }
    if(!ans)
        cout<<"draw";
    else
        cout<<ans;
    return 0;
}