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);	
	}
}

 

上一篇:转载:你可以杀我,但你不能评价(judge)我


下一篇:程序员面试的刷题网站都总结在这了,想要的赶紧拿走!