CDOJ1927 爱吃瓜的伊卡洛斯(2) 【并查集】启发式合并+set

伊卡洛斯很爱吃西瓜。一次，他来到一个西瓜摊旁，发现水果摊有N个西瓜，西瓜有红色、黄色、绿色、蓝色……等等数不清的颜色。 伊卡洛斯很想知道知道一些信息，便于老板交谈了起来。 当老板的话的第一个字符为”A”时，老板会告诉伊卡洛斯一些信息，格式如下： A x y 1 这句话表示第x个西瓜和第y个西瓜是同一种颜色的。 A x y 2这句话表示第x个西瓜和第y个西瓜是不同种颜色的。

当然，为了考验伊卡洛斯有没有认真听， 老板也会时不时问伊卡洛斯一些问题，格式如下： Q x y 这句话表示询问第x个西瓜和第y个西瓜是不是同一种颜色，如果确定为同一种颜色，伊卡洛斯需要回答1；确定为不同种颜色，伊卡洛斯需要回答2；无法确定时伊卡洛斯回答3。 注意，伊卡洛斯是根据已获得的信息来回答的。也就是只有这个问题之前的信息才为已知信息。

老板说，只有回答对他全部的问题，伊卡洛斯才能吃到瓜，他聪明的想到了让你来帮助他。

Input

第一行包含两个整数N和M，N是西瓜总数，M是以A或Q开头的老板的话总和。
以下M行，每行包含一条老板的话。形式有A x y 1或A x y 2或Q x y。 1≤N≤100000 1≤M≤200000 1≤X,Y≤N 数据保证没有矛盾

Output

对于每一条Q指令，输出1/2/3代表两个西瓜颜色的关系。

Hint

西瓜的颜色可以有无数多种！

思路：这题颜色有无穷多种，所以不能按照普通的并查集分块求对立。 
可以用set存每一个集合的对立（不同颜色）集合，如果两个x，y颜色相同时，合并两个集合的对立集合。 
合并的时候尽量将小的集合合并到大的集合中去。

 #include <algorithm>
 #include <iostream>
 #include <cstdio>
 #include <cmath>
 #include <cstring>
 #include <set>
 using namespace std;

 const int N = 1e5+;
 int f[N];
 set<int> s[N];

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

 void unite(int x,int y)
 {
     x = find(x);
     y = find(y);
     if(x!=y)
     {
         if(s[x].size()>s[y].size())swap(x,y);
         // s[x] size is small
         f[x]=y;
         if(s[x].size())
         {
             set<int>:: iterator it;
             for(it = s[x].begin(); it != s[x].end(); it++)
                 s[y].insert(find(*it));
         }
     }
     return;
 }

 int main()
 {
     int n,m;
     char op[];
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++)
         f[i]=i;
     while(m--)
     {
         scanf("%s",&op);
         int x,y,z;
         if(op[] =='A')
         {
             scanf("%d%d%d",&x,&y,&z);
             if(z==)
                 unite(x,y);
             else
             {
                 s[find(x)].insert(find(y));
                 s[find(y)].insert(find(x));
             }
         }
         else
         {
             scanf("%d%d",&x,&y);
             int xx = find(x);
             int yy = find(y);
             if(xx==yy)printf("1\n");
             else if(s[xx].find(yy)!=s[xx].end() || s[yy].find(xx)!=s[yy].end())
                 printf("2\n");
             else printf("3\n");
         }
     }
     return ;
 }