信息奥赛一本通 并查集 团伙(group)
原创
©著作权归作者所有:来自51CTO博客作者mb5f5b1df7f1e34的原创作品,请联系作者获取转载授权,否则将追究法律责任
团伙(group)
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 596 通过数: 247
【题目描述】
在某城市里住着n个人,任何两个认识的人不是朋友就是敌人,而且满足:
1、我朋友的朋友是我的朋友;
2、我敌人的敌人是我的朋友;
所有是朋友的人组成一个团伙。告诉你关于这n个人的m条信息,即某两个人是朋友,或者某两个人是敌人,请你编写一个程序,计算出这个城市最多可能有多少个团伙?
【输入】
第1行为n和m,1<n<1000,1≤m≤100 000;
以下m行,每行为p x y,p的值为0或1,p为0时,表示x和y是朋友,p为1时,表示x和y是敌人。
【输出】
一个整数,表示这n个人最多可能有几个团伙。
【输入样例】
6 4
1 1 4
0 3 5
0 4 6
1 1 2
【输出样例】
3
【算法分析】
我们可以把朋友的节点设为一样,记录在fa[i]中,遇到敌人时,将敌人的敌人放到fa[i]中,所以需要a[i]记录敌人。
【代码实现】
#include<bits/stdc++.h>
using namespace std;
#define N 10100
int fa[N],a[N];
//fa[i]记录父亲节点:朋友,a[i]记录i的第一个敌人
int find(int x)
{
return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=2*n;i++)
fa[i]=i;
for(int i=1;i<=m;i++)
{
int q,x,y;
scanf("%d%d%d",&q,&x,&y);
int r1=find(x),r2=find(y);
if(q==0)
{
fa[r2]=r1;
}
else
{//如果是敌人,那么合并x与y的敌人,y也如此
if(!a[x]) //如果x无敌人,那么就把y当作他的敌人
a[x]=y;
else
{
int r3=find(a[x]);
fa[r3]=r2; //敌人的敌人是朋友
}
//y同理
if(!a[y])
a[y]=x;
else
{
int r4=find(a[y]);
fa[r4]=r1;
}
}
}
int tot=0;
for(int i=1;i<=n;i++)
if(fa[i]==i) tot++;
cout<<tot<<endl;
return 0;
}