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
;
set<int> s
;

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\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 0;
}