【解题思路】
每个人是一个元素,一个团伙是一个集合。
如果A与B是朋友,那么A与B同属于一个集合,应该将A所在的集合与B所在的集合合并。
如果A与C是敌人,A与D是敌人,根据“我敌人的敌人是我的朋友”,A是C的敌人,D是C的敌人(A)的敌人,因此D与C是朋友。同理,A的所有敌人都是朋友,只需要知道A的一个敌人,通过并查集中的查询操作,就能够知道A的敌人组成的集合是什么。也就是说一个人的所有敌人都互为朋友,同属于一个集合。
因此,设数组enemy, enemy[i]表示i的某一个敌人。
当x与y互为敌人时,先考虑y作为x的敌人的情况:
- 如果x还没有敌人(enemy[x] == 0),就将y设为x的一个敌人,即enemy[x]=y
- 如果x已有敌人(enemy[x] > 0),那么应该将y加入x的敌人组成的集合,即将y与enemy[x]所在的集合合并。
反过来,同时要考虑x作为y的敌人的情况:
- 如果y还没有敌人(enemy[y] == 0),就将x设为y的一个敌人,即enemy[y]=x
- 如果y已有敌人(enemy[y] > 0),那么应该将x加入y的敌人组成的集合,即将x与enemy[y]所在的集合合并。
并查集中每个集合的根结点满足fa[i] == i,集合的数量就是fa数组中根结点的个数。最后输出集合的数量。
【参考代码】
//示例代码
#include <bits/stdc++.h>
using namespace std;
int n,m,p,x,y,ans;
int father[1010],enemy[1010];
int find(int x){//查找根
if(father[x]!=x) father[x]=find(father[x]);
return father[x];
}
void unionn(int x,int y){//合并
int a=find(x);
int b=find(y);
father[a]=b;
}
int main() {
cin>>n>>m;
for(int i=1;i<=n;i++) father[i]=i;//初始化每个人都是自己的团伙头
for(int i=1;i<=m;i++){
cin>>p>>x>>y;
if(p==0) unionn(x,y);//朋友合并
else {
if(enemy[x]==0) enemy[x]=y;//无敌人则建立敌人
else unionn(enemy[x],y);//已有敌人 敌人和敌人是朋友
if(enemy[y]==0) enemy[y]=x;//无敌人则建立敌人
else unionn(enemy[y],x);//已有敌人 敌人和敌人是朋友
}
}
for(int i=1;i<=n;i++)
if(father[i]==i) ans++;//统计每个团伙的头有几个
cout<<ans;
return 0;
}