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+6;
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[6];
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        f[i]=i;
    while(m--)
    {
        scanf("%s",&op);
        int x,y,z;
        if(op[0] =='A')
        {
            scanf("%d%d%d",&x,&y,&z);
            if(z==1)
                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
");
            else if(s[xx].find(yy)!=s[xx].end() || s[yy].find(xx)!=s[yy].end())
                printf("2
");
            else printf("3
");
        }
    }
    return 0;
}