1385:团伙(group)

【解题思路】
每个人是一个元素,一个团伙是一个集合。
如果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;
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值