[置顶] 小米2013年校园招聘笔试题-简单并查集

九度:http://ac.jobdu.com/problem.php?pid=1526

题目描述:

假如已知有n个人和m对好友关系(存于数字r)。如果两个人是直接或间接的好友(好友的好友的好友...),则认为他们属于同一个朋友圈,请写程序求出这n个人里一共有多少个朋友圈。
假如:n = 5 , m = 3 , r = {{1 , 2} , {2 , 3} , {4 , 5}},表示有5个人,1和2是好友,2和3是好友,4和5是好友,则1、2、3属于一个朋友圈,4、5属于另一个朋友圈,结果为2个朋友圈。

输入:

输入包含多个测试用例,每个测试用例的第一行包含两个正整数 n、m,1=<n,m<=100000。接下来有m行,每行分别输入两个人的编号f,t(1=<f,t<=n),表示f和t是好友。 当n为0时,输入结束,该用例不被处理。

输出:

对应每个测试用例,输出在这n个人里一共有多少个朋友圈。

样例输入:
5 3
1 2
2 3
4 5
3 3
1 2
1 3
2 3
0
样例输出:
2
1

#include <stdio.h>
int n, m, f[100001],i;

int findSet(int x) {
	int t = x;
	while (f[x] > 0)
		x = f[x];
	while (t != x) { //路径压缩
		int q = f[t];
		f[t] = x;
		t = q;
	}
	return x;
}

void unionSet(int x, int y) {
	int fx = findSet(x);
	int fy = findSet(y);
	if (fx != fy) {
		n--;
		int tmp = f[fx] + f[fy];//两个集合个数之和
		if (f[fx] < f[fy]) {  //大集合 吞并小集合.
			f[fy] = fx;
			f[fx] = tmp;
		} else {
			f[fx] = fy;
			f[fy] = tmp;
		}
	}
}

int main() {
	//freopen("in.txt", "r", stdin);
	int a,b;
	while(scanf("%d",&n), n) {
		scanf("%d", &m);
		for( i=1; i<=n; i++)
		f[i] = -1;
		for(i=0; i<m; i++) {
			scanf("%d %d", &a,&b);
			unionSet(a,b);
		}
		printf("%d\n", n);
	}
	return 0;
}

http://www.acmerblog.com/offer-xiaomi-1167.html




你可能感兴趣的:([置顶] 小米2013年校园招聘笔试题-简单并查集)