现在的位置: 首页 > 综合 > 正文

【互联网面试】朋友圈问题

2013年08月22日 ⁄ 综合 ⁄ 共 838字 ⁄ 字号 评论关闭

问题描述:

假如已知有n个人和m对好友关系(存于数字r)。如果两个人是直接或间接的好友(好友的好友的好友...),

则认为他们属于同一个朋友圈,

请写程序求出这n个人里一共有多少个朋友圈。

假如:n = 5 m = 3
r = {{1 , 2} , {2 , 3} ,{4 , 5}}
,表示有5个人,12是好友,23是好友,45是好友,

123属于一个朋友圈,45属于另一个朋友圈,结果为2个朋友圈。

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

 据说用并查集来解决。。。。看不懂代码……

 

//朋友圈问题

#include <iostream>
using namespace std;

int set[10001];
//带路径优化的并查集查找算法
inline int find(int x)             
{  
	int i,j,r; 
	r = x;  
	while(set[r] != r)   
	{
		r = set[r];  
	}
	i = x;
	while(i != r)
	{
		j = set[i];
		set[i] = r;
		i = j;
	}  
	return r;  
} 
//优化的并查集归并算法
inline void merge(int x, int y)       
{  
    int t = find(x);  
    int h = find(y);  
    if(t < h)  
	{
		set[h] = t;  
	}
    else  
	{
		set[t] = h;  
	}
}

int friends(int n , int m , int r[][2])  
{  
    int i , count;  
	//初始化并查集,各点为孤立点,分支数为n
    for(i = 1 ; i <= n ; ++i)       
	{
		set[i] = i;  
	}
    for(i = 0 ; i < m ; ++i)  
	{
		merge(r[i][0] , r[i][1]);  
	}
    for(i = 1 ; i <= n ; ++i)       
	{
		cout << set[i] <<" ";  
	}
    count = 0;  
    for(i = 1 ; i <= n ; ++i)  
    {  
        if(set[i] == i)  
		{
			++count;  
		}
    }  
    return count;  
}

void main()
{
	int n=5;
	int m=3;
	int a[][2]={{1,2},{2,3},{4,5}};
	cout << friends(n,m,a) <<endl;
}

 

抱歉!评论已关闭.