并查集-判断是否存在环

不论是向下连一条边，还是向右连一条边，只要新连的那个点也在边所在的这个集合中，那么就成环了。如果不在这个集合中，就把这个点加到这个集合中。询问这样的操作m次之后，有没有成环。

/*
    就是一个裸的并查集，不论是向下连一条边，还是
    向右连一条边，只要新连的那个点也在边所在的
    这个集合中，那么就成环了。
*/
#include<iostream>
#include<cstring>

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(x != p[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 res = 0;
    for(int i=0; i<m; i++)
    {
        int x, y;
        int pa, pb;
        char s;
        
        cin >> x >> y >> s;
        x--, y--;
        pa = get(x, y);
        
        if(s == 'D')
            pb = get(x+1, y);
        else
            pb = get(x, y+1);
        
        // 我这样find了好几次，直接把find的结果先存下来不好吗？
        int a = find(pa), b = find(pb);
        if(a == b)
        {
            res = i+1;
            break;
        }
        else p[b] = a;
            
    }
    
    if(!res) puts("draw");
    else printf("%d\n", res);
        
    return 0;
}