并查集最简单的模板
#include <bits/stdc++.h>
using namespace std;
int p[100010];
int n,m;
int Find(int x)//返回x的祖宗节点+路径压缩
{
if(p[x]!=x)
p[x]=Find(p[x]);
return p[x];
}
int main()
{
scanf("%d %d",&n,&m);
for(i=1;i<=n;i++)
p[i]=i;
while(m--)
{
char op[2];
scanf("%s%d%d",op,&a,&b);
if(op[0]=='M')
p[Find(a)]=Find(b);//给a的祖宗认了个爹;
else
{
if(Find(a)==Find(b))
puts("Yes");
else puts("No");
}
}
return 0;
}
例题:食物链
动物王国中有三类动物 A,B,CA,B,C,这三类动物的食物链构成了有趣的环形。
AA 吃 BB,BB 吃 CC,CC 吃 AA。
现有 NN 个动物,以 1∼N1∼N 编号。
每个动物都是 A,B,CA,B,C 中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这 NN 个动物所构成的食物链关系进行描述:
第一种说法是
1 X Y
,表示 XX 和 YY 是同类。第二种说法是
2 X Y
,表示 XX 吃 YY。此人对 NN 个动物,用上述两种说法,一句接一句地说出 KK 句话,这 KK 句话有的是真的,有的是假的。
当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
- 当前的话与前面的某些真的话冲突,就是假话;
- 当前的话中 XX 或 YY 比 NN 大,就是假话;
- 当前的话表示 XX 吃 XX,就是假话。
你的任务是根据给定的 NN 和 KK 句话,输出假话的总数。
输入格式
第一行是两个整数 NN 和 KK,以一个空格分隔。
以下 KK 行每行是三个正整数 D,X,YD,X,Y,两数之间用一个空格隔开,其中 DD 表示说法的种类。
若 D=1D=1,则表示 XX 和 YY 是同类。
若 D=2D=2,则表示 XX 吃 YY。
输出格式
只有一个整数,表示假话的数目。
数据范围
1≤N≤500001≤N≤50000,
0≤K≤1000000≤K≤100000输入样例:
100 7 1 101 1 2 1 2 2 2 3 2 3 3 1 1 3 2 3 1 1 5 5
输出样例:
3
#include <bits/stdc++.h>
using namespace std;
const int N=50010;
int n,m;
int p[N],d[N];
int Find(int x)
{
if(p[x]!=x)
{
int t=Find(p[x]);
d[x]+=d[p[x]];
p[x]=t;
}
return p[x];
}
int main()
{
scanf("%d %d",&n,&m);
for(int i=1;i<=n;i++) p[i]=i;
int res=0;
while(m--)
{
int t,x,y;
scanf("%d %d %d",&t,&x,&y);
if(x>n||y>n) res++;
else
{
int px=Find(x);
int py=Find(y);
if(t==1)
{
if(px==py&&(d[x]-d[y])%3) res++;
else if(px!=py)
{
p[px]=py;
d[px]=d[y]-d[x];
}
}
else
{
if(px==py&&(d[x]-d[y]-1)%3) res++;
else if(px!=py)
{
p[px]=py;
d[px]=d[y]+1-d[x];
}
}
}
}
printf("%d\n",res);
return 0;
}