并查集是一种树形的数据结构,用于处理一些不相交集合的合并及查询问题。常常在使用中以森林来表示,进行快速调整。
案例:朋友圈
题目描述:
假如已知有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
分析代码的时间复杂度,空间复杂度,正确率和效率
Int friends(int n,int m,int* r[]);
int set[10001];
int find(int x) //带路径优化的并查集查找算法
{
int i, j, r;
r = x;
while (set[r] != r) //寻找此集合的代表
r = set[r];
i = x;
while (i != r) //使得r代表的集合中,所有结点直接指向人,即路劲压缩
{
j = set[i];
set[i] = r;
i = j;
}
return r;
}
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[])
{
int i,count;
for (i = 1; i <= n; i++)//初始化并查集,各点为孤立点,分支数为n
set[i] = i;
for (i = 0; i < m; i++)//读入r
merge(r[i][0],r[i][1]);
count = 0;
for (i = 1; i <= n; i++)
{
if (set[i] == i)
++count;
}
return count;
}
通过类的方法解决:
#include "vector"
#include "string"
#include "algorithm"
#include <iostream>
#include "stack"
#include "map"
#include <cmath>
using namespace std;
#define MAX 100
class UFSet
{
public:
UFSet()
{
parent = new int[MAX];
};
~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
};
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]);//递归,最终找到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 )
{
if (n == 0) break;
cin >> m;
if (m >= MAX) break;
UFSet uset;
uset.makeSet(n);//初始化
//接收m对关系
int x = 0;
int y = 0;
for (int i = 0; i<m; i++)
{
cin >> x >> y;//注:这里数组下标代表人的对应编号
uset.unionSet(x, y);
}
cout << uset.getSets(n) << endl;
}
return 0;
}