算法笔记---好朋友

题目描述

有一个叫作“数码世界”的奇异空间，在数码世界里生活着许许多多的数码宝贝，其中有些数码宝贝之间可能是好朋友。并且数码世界有两条不成文的规定：
第一，数码宝贝A和数码宝贝B是好朋友等价于数码宝贝B和数码宝贝A是好朋友。
第二，如果数码宝贝A和数码宝贝C是好朋友，而数码宝贝B和数码宝贝C也是好朋友，那么A和B也是好朋友。

现在给出这些数码宝贝中所有好朋友的信息，问：可以把这些数码宝贝分成多少组，满足每组中的任意两只数码宝贝都是好朋友，且任意两组之间的数码宝贝都不是好朋友。

样例：

输入：
4 2
1 4
2 3
输出：
2

输入：
7 5
1 2
2 3
3 1
1 4
5 6
输出：
3

解题思路：
可以用并查集思想来求解
将输入的两个好朋友进行合并，最后不能合并的集合的个数就是分成的组数
注意：

每个集合只有一个根结点

下面的代码除了输出数码宝贝分成的组数，还输出了每个集合的根结点及其对应的子结点

#include<iostream>
#include<vector>
#include<map>
using namespace std;

const int max_union_find_set = 110;
int father[max_union_find_set] = { 0 };

//初始化
void init_father(int n) {//n 表示共有多少个数字
	for (int i = 1; i <= n;i++)
	{
		father[i] = i;
	}
}

//查找x所在集合的父结点
int find_father(int x) {
	int current_temp = x;//current_temp为最初查找结点
	while (x != father[x])
	{
		x = father[x];
	}
	//当前 x 为根结点

	//路径压缩
	while (current_temp != father[current_temp])
	{
		int temp = current_temp;//临时保存当前结点
		current_temp = father[temp];//往上查找
		father[temp] = x;//进行路径压缩
	}
	return x;
}

//合并两个元素到一个集合
void union_father(int a, int b) {
	int father_a = find_father(a);
	int father_b = find_father(b);
	if (father_a != father_b)
	{
		father[father_a] = father_b;
	}
}

int main() {

	
	int n, m;//n 表示共有多少个数字，m 表示好朋友的组数
	cin >> n >> m;
	//初始化 father 数组
	init_father(n);
	int a, b;// a ，b 分别表示输入的具体数字，即 a 和 b 为好朋友
	for (int i = 0; i < m; i++)
	{
		cin >> a >> b;
		union_father(a, b);//合并两个数字
	}
	map<int, vector<int>> root_map;//保存每个根结点及其子结点

	vector<int> result;// 存放根结点

	for (int i = 1;i <= n;i++)
	{
		int root = find_father(i);
		result.push_back(root);
	}

	for (vector<int>::iterator it = result.begin();it != result.end();it++)//得出根结点和每个根结点对应的子节点
	{
		int root = (*it);
		vector<int> child;
		for (int i = 1;i <= n;i++)
		{
			if (father[i] == root && i != root)
			{
				child.push_back(i);
			}
		}
		root_map.insert(pair<int, vector<int>>(root, child));
	}

	cout << "组数为：" << root_map.size() << endl;

	for (map<int,vector<int> >::iterator it = root_map.begin();it != root_map.end();it++)
	{
		cout << "根结点为：" << (*it).first << endl;
		if ((*it).second.size() != 0)
		{
			cout << "其子结点分别为：";
		}
		for (vector<int>::iterator vit = (*it).second.begin();vit !=(*it).second.end();vit++)
		{
			cout << (*vit) << " ";
		}
		cout << endl;
	}
	system("pause");
	return 0;
}

运行结果：