2013年小米校园招聘笔试题---研发

3、朋友圈(25分) 

  假如已知有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个朋友圈。 

  最后请分析所写代码的时间、空间复杂度。评分会参考代码的正确性和效率。

思路：简单的并查集的应用

#include <iostream>        
using namespace std;       
const int MAX_N=1000;      
int N,M;                   
int par[MAX_N],rank[MAX_N];
void initset(int n){
	for(int i=1;i<=n;i++){
		par[i]=i;
		rank[i]=0;
	}
}    
int findpar(int x){
	if(x==par[x]) return x;
	return par[x]=findpar(par[x]);
}    
void unionset(int x,int y){
	x=findpar(x);
	y=findpar(y);
	if(x==y)return ;
	if(rank[x]<rank[y]){
		par[x]=y;
	}
	else{
		par[y]=x;
		if(rank[x]==rank[y]) rank[x]++;
	}
}
int main(){
	while(cin>>N>>M){          // N persons, M pair of friends.
		initset(N);
		for(int i=0;i<M;i++){
			int x,y;
			cin>>x>>y;
			unionset(x,y);
		}
		int res=0;
		for(int i=1;i<=N;i++){
			if(par[i]==i) res++;
		}
		cout<<res<<endl;
	}
	return 0;
}