格子游戏(二维坐标下的并查集)

#include<bits/stdc++.h>
using namespace std;

const int N=40010;
int p[N];
int n,m;

int get(int x,int y)
{
    return x*n+y;
}

int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    
    return p[x];
}

int main()
{
    scanf("%d%d",&n,&m);
    
    for(int i=0;i<n*n;i++) p[i]=i;
    
    int res=0;
   for(int i=1;i<=m;i++)
    {
       int x,y;
       char op[2];

       scanf("%d%d%s",&x,&y,op);

       x--,y--;
       
        int a=get(x,y), pa=find(a),b;
       
       if(*op=='D') b=get(x+1,y);
       else b=get(x,y+1);
       
       int pb=find(b);
       
       if(pa==pb)
         {
            res=i;
            break;
         }
        else p[pa]=pb;
    }
    
    if(!res) puts("draw");
    else printf("%d\n",res);
    
    return 0;
}