poj1611 The Suspects (并查集)

题意：题目很长...总结下来就是 输入人个数n(编号0~n-1),  有m组信息(m个例子)，其中默认标号0的为嫌疑人， 和0一组的都为嫌疑人，然后确定嫌疑人个数

思路：集合运算，就是将与0相关联的(可能式间接关联)放在一个集合中然后统计人数.所以利用并查集的Union,find,judge函数来统计个数。即最后用judge函数判断是否与0在同一个集合中，如果是则ans++；

完整代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#define maxn 30005
#define qmax 500
int pre[maxn];
int n,m,ans;

int find(int x){
	if(x == pre[x]) return pre[x];
	return pre[x] = find(pre[x]);
}
bool judge(int x,int y){
	if(find(x)==find(y)) return true;
	else return false;
}
void Union(int x,int y){
	x = find(x);
	y = find(y);
	if(x==y) return;
	//将x加入y集合中 
	pre[x] = y;
	
}
void init(){
	ans = 0;
	for(int i = 0;i<n;i++){
		pre[i] = i;
	}
}
int main(){

	while(~scanf("%d%d",&n,&m)){
		if(n==0&&m==0) break;
		init(); 
		for(int i = 0;i<m;i++){
			int k;
			scanf("%d",&k);
			if(k==1){
				int l;
				scanf("%d",&l);
			}
			else{
				int pr,nex;
				for(int i = 0;i<k;i++){
					if(i == 0) {
						scanf("%d",&pr);
					}else{
						scanf("%d",&nex);
						Union(pr,nex);
						pr = nex;
					}
				}
			}	
		} 
		for(int i = 0;i<n;i++){
			if(judge(0,i)) ans++;
		}
		printf("%d\n",ans);	
	}
}

标签：pre,Suspects,return,int,查集,ans,judge,poj1611,find
来源： https://blog.csdn.net/qq_42825732/article/details/95643931