0802-并查集-BZOJ1370 Gang团伙(水题)

传送门

由于题目描述清晰易懂,我就不用转述啦,大家自己打开看看就是了

(只是我觉得 input 可能写的不是很清楚,应该是E-->enemy 敌人,F-->friend 朋友)

题解:

这是一道很并查集的并查集的题了,直接把是朋友的合并在一起,把敌人的敌人和自己合并在一起。大致操作就是用一个en [ i ]  数组来表示  i 的上一个敌人是谁,每次读入到它的下一个敌人就更新 en 数组,然后搞一搞就出来了

代码:

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#define N 1009
using namespace std;
int n,m,fa[N],en[N];
int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
void merge(int x,int y){
	int fx=find(x),fy=find(y);
	if(fx==fy) return;
	fa[fx]=fy;
}
int main(){
	scanf("%d%d",&n,&m);
	int i,j,k,x,y;
	char st;
	for(i=1;i<=n;++i) fa[i]=i;
	for(i=1;i<=m;++i){
		cin>>st;
		scanf("%d%d",&x,&y);
		if(st=='E') {
			if(en[x])	merge(y,en[x]);//因为一开始 en 数组全为0
			if(en[y])	merge(x,en[y]);
			en[x]=y;en[y]=x;//更新
		}
		else	merge(x,y);
	}
	int ans=0;
	for(i=1;i<=n;++i){
		int f=find(i);
		if(f==i) ans++;
	}
	printf("%d",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_42557561/article/details/81359501