数据结构（并查集+set)

L - 爱吃瓜的伊卡洛斯(2)

 

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

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

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

Input

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

Output

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

Sample Input

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

Sample Output

2
3
3  
1

Hint

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

题解：因为西瓜有无数种不能像只有两种一样如果x与y不同,y与z不同那么x与z就相同那样来做题，这里我们需要用set来存放与x不同的y，与y不同的x,并且都是对它们的根节点操作，因为x根节点与y的根节点不同那么x,y也不相同。

当fnd(x)==fnd(y)时表示相同，当fnd(x)!=fnd(y)时如果x的根节点的set集合里面有的y根节点，或y的根节点得set集合里面有x的根节点就表明x,y不同，否则不能判断。。。

AC代码：

#include<iostream>
#include<string>
#include<cstring>
#include<set>
#include<cstdio>
using namespace std;
const int maxn=100005;
set<int>s[maxn];
int parent[maxn];
int fnd(int x)
{
    return parent[x]!=x?parent[x]=fnd(parent[x]):x;
}
bool uni(int x,int y)
{
    int fx=fnd(x),fy=fnd(y);
    if(fx!=fy)
    {
        if(s[x].size()>s[y].size()) swap(s[x],s[y]);//保证s[x]小，s[y]大
        parent[fx]=fy;
        for(set<int>::iterator it=s[x].begin();it!=s[x].end();++it)
        {
            s[y].insert(fnd(*it));//把s[x]中所有与x不同的西瓜的根节点与s[y]合并
        }
        return true;
    }
    return false;
}
int main()
{
    int n,m;
    //ios::sync_with_stdio(0);cin.tie(0);
    cin>>n>>m;
    getchar();
    for(int i=1;i<=n;i++)
        parent[i]=i;
    char op;
    int x,y,p;
    while(m--)
    {
        cin>>op>>x>>y;
        if(op=='A')
        {
            cin>>p;
            if(p==1)
            {
                uni(x,y);
            }
            else
            {
              x=fnd(x);
              y=fnd(y);
              s[x].insert(y);
              s[y].insert(x);
            }
        }
        else
        {
            x=fnd(x);y=fnd(y);
            if(x==y)
                cout<<1<<endl;
            else if(s[x].count(y)==1||s[y].count(x)==1)//x里有y或y里有x都可确定x,y不同
                cout<<2<<endl;
            else cout<<3<<endl;
        }
    }
    return 0;
}