AcWing 837. 连通块中点的数量(并查集)

题目链接 ： 点击查看

题目描述 ：

给定一个包含 n 个点（编号为 1∼n）的无向图，初始时图中没有边。

现在要进行 m 个操作，操作共有三种：

1. C a b，在点 a 和点 b 之间连一条边，aa 和 bb 可能相等；
2. Q1 a b，询问点 a 和点 b 是否在同一个连通块中，a 和 b 可能相等；
3. Q2 a，询问点 a 所在连通块中点的数量；

输入输出格式 ：

输入

第一行输入整数 n 和 m。

接下来 m 行，每行包含一个操作指令，指令为 C a b，Q1 a b 或 Q2 a 中的一种。

输出 

对于每个询问指令 Q1 a b，如果 a 和 b 在同一个连通块中，则输出 Yes，否则输出 No。

对于每个询问指令 Q2 a，输出一个整数表示点 a 所在连通块中点的数量

每个结果占一行。

输入输出样例 ：

输入

5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5

输出

Yes
2
3

题目分析 ：

本题表面上是图论问题，实际上我们可以用并查集解决，具体做法与合并集合这道题类似，只不过这道题所维护的并查集包含了一个cnt[N]数组，用来记录每个集合的点数。要解决这道题，首先要对并查集数组进行初始化，初始化时每一个元素都是以自己为根节点构成的集合，每个集合的元素都是1，即p[i] = i, cnt[i] = 1(这里p数组主要用来记录每个节点的父节点，根节点的父节点就是他本身)。接着，我们看一下每种操作，'C'操作是将点a 与 点 b 之间连一条边，其实就是数组a 与数组 b所在集合合并，这是构成并查集的基本操作，其间通过一个find函数来查找a点所在集合的根节点，中间用到了路径压缩算法，把每个节点直接与根节点相连，在此就不赘述了。并且我们规定每个节点所在集合的根节点cnt数组，记录的才是整个集合点的个数。将两个集合合并要将集合a作为子树接到集合b下，在将两个集合拥有的点相加。如此，‘C’操作的具体做法为p[find(a)] = find(b), cnt[find(b)] += cnt[find(a)]。‘Q1’操作是判断 点 a 与点 b是否在同一集合中，即判断find(a) == find(b)是否为真即可。'Q2‘操作是输出a点所在集合点的个数 即输出 cnt[find(a)]即可。详见如下代码。

代码 ：

#include<iostream>
#include<cstdio>
using namespace std;
const int N = 1e5 + 7;
int n, m;
int p[N], cnt[N];
int find(int x) {
	if (p[x] != x) p[x] =find(p[x]);//路径压缩 
	return p[x];
}

int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i ++ )  {
		p[i] = i;//初始时 每个集合的根节点为其本身 
		cnt[i] = 1;//初始时 每个集合数量均为1 
	}
	while (m -- ) {
		string op;
		int a, b;
		cin >> op;
		if (op == "C") {
		    cin >> a >> b;
            a = find(a), b = find(b);
            if (a != b) {
                p[a] = b;
                cnt[b] += cnt[a];
            }
		}
		else if (op == "Q1") {
			cin >> a >> b;
			if (find(a) == find(b)) cout << "Yes" << endl;
			else cout << "No" << endl; 
		}
		else {
			cin >> a;
			cout << cnt[find(a)] << endl;
		}
	} 
	return 0;
} 

-------------------------------------------------------------------------

下面给出本题的相关模板

维护size的并查集：

    int p[N], size[N];
    //p[]存储每个点的祖宗节点, size[]只有祖宗节点的有意义，表示祖宗节点所在集合中的点的数量

    // 返回x的祖宗节点
    int find(int x)
    {
        if (p[x] != x) p[x] = find(p[x]);
        return p[x];
    }

    // 初始化，假定节点编号是1~n
    for (int i = 1; i <= n; i ++ )
    {
        p[i] = i;
        size[i] = 1;
    }

    // 合并a和b所在的两个集合：
    size[find(b)] += size[find(a)];
    p[find(a)] = find(b);