-
题目描述:
-
假如已知有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
分析:利用并查集联立m对朋友,最后检查有多少个孤立集合即可
#include "vector"
#include "string"
#include "algorithm"
#include <iostream>
#include "stack"
#include <cmath>
using namespace std;
class UFSet
{
public:
UFSet(int nsize)
{
size = nsize;
parent = new int[size];
};
~UFSet()
{
delete[] parent;
parent = NULL;
};
void makeSet(int n);初始化每个元素的祖先
int findSet(int x);//找到元素x的祖先元素
void unionSet(int a, int b);//若两个元素的祖先不同,则将x元素的祖先设置为y元素的祖先
int getSets(int n);//获取集合数量
private:
int *parent;//存放祖先节点,例如x=parent[i],元素i的祖先节点为元素x
int size;
};
void UFSet::makeSet(int n) //初始化
{
//初始化每一个元素都各自为一个独立的集合,其祖先均设定为自身
for (size_t i = 1; i <= n; i++)
parent[i] = i;
}
int UFSet::findSet(int x)
{
//找到元素所在的集合,也就是找到自己的最高的祖先,
//这也是判断两个元素是否在同一个集合中的主要依据。
if (parent[x] == x)
return x;
parent[x] = findSet(parent[x]);//找到祖先,而不是父节点
return parent[x];
}
void UFSet::unionSet(int x, int y)
{
//将x和y所在的集合进行合并,利用findSet()判断x和y所在的集合是否相同,
//如果不同,则要把其中一个元素的祖先指向另一个元素的祖先。
int ux = findSet(x);//获取节点x的祖先
int uy = findSet(y);
if (ux != uy)
parent[ux] = uy;
}
int UFSet::getSets(int n)
{
int count = 0;
for (int i = 1; i <= n; i++)
{//如果存在某一个节点的祖先是自身说明他是孤立的或者本身就是祖先
if (parent[i] == i)
count++;
}
return count;
}
int main()
{
int m, n;
while (cin >> n >> m)
{
UFSet uset(100000);
uset.makeSet(n);//初始化
//接收m对关系
int x = 0, y = 0;
for (int i = 0; i<m; i++)
{
cin >> x >> y;
uset.unionSet(x, y);
}
cout << uset.getSets(n) << endl;
}
return 0;
}
/**************************************************************
Problem: 1526
User: EbowTang
Language: C++
Result: Accepted
Time:550 ms
Memory:1912 kb
****************************************************************/
注:本博文为EbowTang原创,后续可能继续更新本文。如果转载,请务必复制本条信息!
原文地址:http://blog.csdn.net/ebowtang/article/details/50484031
原作者博客:http://blog.csdn.net/ebowtang