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