你到底有多少个朋友圈?

    有这样一道面试题它是这样描述的:假设有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属于同一个朋友圈.结果为两个朋友圈.

    如何去分析这道面试题呢？这就用到并查集这种数据结构.

并查集>

   将N个不同的元素分成一组不相交的集合，开始时，每个元素就是一个集合，然后按规律将两个集合进行合并,在合并的过程中要不断的查找这个元素属于哪个集合.

   


    在统计朋友圈的数量的时候只需要统计负数的个数即可.

    #pragma once

class UnionSet
{
public:
UnionSet(int size)
:_size(size+1)
{
_a=new int[size+1];
memset(_a,-1,sizeof(int)*(size+1));
}
~UnionSet()
{
delete[]_a;
_size=0;
}
void Merge(int x1,int x2)
{
int root1=_GetRoot(x1);
int root2=_GetRoot(x2);
if(root1 != root2)
{
_a[root1] += _a[root2];
_a[root2] = root1;
}
}
int CountFriends()
{
int count=0;
for(int i=0;i<_size;++i)
{
if(_a[i] < 0)
{
count++;
}
}
return count-1;
}
protected:
int _GetRoot(int index)
{
int root=index;
while(_a[root] >= 0)
{
root=_a[root];
}
return root;
}
protected:
int *_a;
int _size;
};

void testUnionSet()
{
const int n=5;
const int m=4;
int a[m][2]={{1,2},{2,3},{4,5},{1,3}};
UnionSet unionset(n);
for(int i=0;i<m;++i)
{
unionset.Merge(a[i][0],a[i][1]);
}
cout<<"Friends:"<<unionset.CountFriends()<<endl;
}

   在这里就分享结束啦~~~