拓展域并查集——爱吃瓜的伊卡洛斯(1) CDOJ1926

                                       爱吃瓜的伊卡洛斯(1)

Time Limit: 1000 MS     Memory Limit: 64 MB

伊卡洛斯很爱吃西瓜。一次，他来到一个西瓜摊旁，发现水果摊有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代表两个西瓜颜色的关系。

Sample input and output

Sample Input	Sample Output
6 9 A 1 2 1 A 1 3 1 A 1 4 2 Q 2 4 Q 1 6 A 3 6 1 A 4 5 2 Q 1 5 Q 1 6	2 3 1 1

 

重点是只有两种颜色，所以A与B不同，B与C不同可以得到， A与C相同

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn=100009;
int pre[maxn*2];//x的同类，x的异类
int find(int x) 
{
    int root=x;
    while(root!=pre[root]) 
    {
        root=pre[root];
    }
    int i=x;
    int farther;
    while(pre[i]!=i)    
    {
        farther=pre[i];      
        pre[i]=root;    
        i=farther;      
    }
    return root;
}
bool join(int a,int b) 
{
    int ra=find(a);
    int rb=find(b);    
    if(ra!=rb)           
    {
        pre[rb]=ra;
        return true;
    }
    return false;
}

int main()
{
    int n,m;
    int t,x,y;
    char a[3];
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n*2;++i)
            pre[i]=i;
        while(m--)
        {
            scanf("%s%d%d",a,&x,&y);
            if(a[0]=='A')
            {
                scanf("%d",&t);
                if(t==1)
                {
                    join(x,y);//x与y是同类
                    join(x+n,y+n);//x的异类与y的异类是同一类
                }
                else if(t==2)
                {
                    join(y+n,x);
                    join(y,x+n);
                }
            }
            else if (a[0]=='Q')
            {
                int rx=find(x);
                int ry=find(y);
                int rrx=find(x+n);
                int rry=find(y+n);
                if(rx==ry) cout<<1<<endl;
                else if(rrx==ry||rry==rx) cout<<2<<endl;
                else cout<<3<<endl;
            }

        }

    }
    return 0;
}