POJ1703Find them, Catch them 【种类并查集】

题目链接：http://poj.org/problem?id=1703

题目大意：给n个人,m次询问。A代表询问a, b之间的关系，D代表给出a, b属于不同的帮派。

我的想法：

太菜了，上课的时候没想到这种方法。

思路：

1.感觉有点像2-sat的思想，对于D给出的a, b两人属于不同帮派。我们就将a于b的对立点用并查集维护在一个集合中，也将b于a的对立点连通。

2.查询时，若a,b两人pre值相等则属于同一帮派。若a与b的对立点属于同一集合，或b与a的对立点属于同一集合，则说明a与b属于不同帮派。若上述都不成立则说明关系还未知。

3.对立点是虚构的，不能影响到原来的点。所以数据范围翻倍，x + n代表x的对立点。

注意：不能用 cin 输入,会超时。只能用scanf

 #include<stdio.h>
 const int maxn = 2e5 + 1e4;

 int pre[maxn];

 int find(int x)
 {
     if(pre[x] == x)
         return x;
     else
     {
         int root = find(pre[x]);
         pre[x] = root;
         return pre[x];
     }
 }

 int main()
 {
     int T;
     scanf("%d", &T);
     while(T --)
     {
         int n, m;
         scanf("%d%d", &n, &m);
         getchar();
         for(int i = ; i <=  * n; i ++)
         {
             pre[i] = i;
         }
         for(int i = ; i <= m; i ++)
         {
             char ch;
             int a, b;
             scanf("%c%d%d", &ch, &a, &b);
             if(ch == 'D')
             {
                 int x = find(a), y = find(b);
                 int xx = find(a + n), yy = find(b + n);
                 if(x != yy)
                     pre[yy] = x;
                 if(y != xx)
                     pre[xx] = y;
             }
             if(ch == 'A')
             {
                 int x = find(a), y = find(b);
                 int xx = find(a + n), yy = find(b + n);
                 if(x == y)
                     printf("In the same gang.\n");
                 else if(x == yy || y == xx)
                     printf("In different gangs.\n");
                 else
                     printf("Not sure yet.\n");
             }
             getchar();
         }
     }
     return ;
 }

POJ1703