[C++]天梯赛2017L2-4 部落 解题思路及详解

[原题]
在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。

-输入格式-

输入在第一行给出一个正整数N（≤104），是已知小圈子的个数。随后N行，每行按下列格式给出一个小圈子里的人：

K P[1] P[2] ⋯ P[K]

其中K是小圈子里的人数，P[i]（i=1,⋯,K）是小圈子里每个人的编号。这里所有人的编号从1开始连续编号，最大编号不会超过104。

之后一行给出一个非负整数Q（≤104），是查询次数。随后Q行，每行给出一对被查询的人的编号。

-输出格式-

首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询，如果他们属于同一个部落，则在一行中输出Y，否则输出N。

-输入样例-

4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7

-输出样例-

10 2
Y
N

[解题思路]

根据题意，我们不但需要统计总人数和部落个数，还需要查询两个人是否在同一部落中，这就需要我们为这些个体构建复杂的关系网，从而化繁为简。对于这种需求，我们通常选用并查集的方法。

接下来让我们来简单了解一下并查表及其实现。并查表是一种树形高级数据结构，顾名思义，并查表拥有两个主要的功能：合并(merge)和查找(find)。合并是使两个数据单位产生联系，合并入一个集体中；而查找则是查询某个数据所在集体的编号（一般情况下，我们用根节点的值代表一个整体）。

当然，除此以外，我们还需要一个数组arr[n]来存放数据，同时，我们还需求这个数组能反映出数据之间的关系，因此我们将用如下形式对数据进行管理和存储：arr[数据值]=对应父节点数据值。

在使用这个数组前，我们需要对数组进行初始化，一般采用的方式为：将每个数据看作是一个独立的集体，即认为每个数据节点的根节点是自己。

代码如下：

void init()
{
	for (int i = 0; i < 10010; i++)
	{
		arr[i] = i;
	}
}

接下来，让我们来实现find函数的功能。其原理也并不复杂：当传入的参数x本身为根节点数值时，直接返回x；否则，我们调用函数递归，返回根节点的数据值。

其实现如下：

int find(int x)
{
	if (arr[x] == x) return x;
	return find(arr[x]);
}

实际上，这种写法存在一个很大的缺陷，即每次寻找数据的根节点值，都要经过一次完整的递归搜索，如果是一个层数较高的树形结构，这样做无疑会产生巨大的时间损耗，因此我们需要对这个过程进行路径压缩。

对于一个复杂的树形结构，我们期望进行如下的处理：

 这样，无论数据原本处于第几层，最后都与根节点直接相连，因此，判断数据处于哪一个集体中仅需经过一次递归。因此，我们可以在数据第一次查找成功后，将其数组直接指向根节点数据。

相应代码进行如下修改：

int find(int x)
{
	if (arr[x] == x) return x;
	return arr[x] = find(arr[x]);
}

接着，让我们对merge函数进行编写。我们预期将两个不同集体（记根节点分别为A,B）中的数据合并为一个集体，那我们不妨使A成为B的根节点，或使B成为A的根节点，这个过程如下图所示：

代码如下：

void merge(int x, int y)
{
	x = find(x);
	y = find(y);
	arr[x] = y;
}

这样，我们就将整个并查集的功能函数完成了。

接下来，我们就可以将并查集应用在这道题上了，不断将在同一个部落的人合并，最后进行查找输出即可。此外，我们还需要统计总人数和部落数，我们可以选择unordered_set容器进行查重。

最后，让我们来看一下这道题的完整代码：

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

//全局变量arr的声明
int arr[10010];

void init()
{
	for (int i = 0; i < 10010; i++)
	{
		arr[i] = i;
	}
}

int find(int x)
{
	if (arr[x] == x) return x;
	return arr[x] = find(arr[x]);
}

void merge(int x, int y)
{
	x = find(x);
	y = find(y);
	arr[x] = y;
}

int main()
{
	vector<int> circles;//用第一个人的编号代替这个圈子，存放每一个圈子
	unordered_set<int> Population, Tribes;//用set容器查重，统计总人数和部落数

	//初始化数组
	init();

	//读入初始数据部分
	int N;
	cin >> N;
	while (N--)
	{
		int K;
		cin >> K;
		int first;
		cin >> first;//每个圈子首先读取第一个人的编号，作为并查集的根节点
		Population.emplace(first);
		circles.push_back(first);
		K--;
		while (K--)
		{
			int other;
			cin >> other;
			Population.emplace(other);
			merge(first, other);//将其他人与第一个人合并
		}
	}

	//将circles中的数据分别寻找各自的根节点，并放入Tribes中查重，即可得到部落数
	for (int i : circles)
	{
		Tribes.emplace(find(i));
	}

	//分别打印总人数及部落数
	cout << Population.size() << ' ' << Tribes.size() << endl;

	//查询部分
	int Q;
	cin >> Q;
	while (Q--)
	{
		int a, b;
		cin >> a >> b;
		if (find(a) == find(b))
			cout << 'Y';
		else
			cout << 'N';
		if (Q != 0)
			cout << '\n';//行末不输出换行
	}
	return 0;
}