7.13日训练 —— 并查集

staygone

已于 2022-07-13 21:26:01 修改

阅读量210

点赞数 2

分类专栏： c++集训记录文章标签：大数据

于 2022-07-13 16:46:15 首次发布

本文链接：https://blog.csdn.net/Gaim_/article/details/125766744

版权

c++集训记录专栏收录该内容

9 篇文章 2 订阅

订阅专栏

并查集的概念：

啊

复制————

并查集(Disjoint-Set或Union-Find Set)是一种表示不相交集合的数据结构，用于处理不相交集合的合并与查询问题。在不相交集合中，每个集合通过代表来区分，代表是集合中的某个成员，能够起到唯一标识该集合的作用。一般来说，选择哪一个元素作为代表是无关紧要的，关键是在进行查找操作时，得到的答案是一致的(通常把并查集数据结构构造成树形结构，根节点即为代表)。在不相交集合上，需要经常进行如下操作:

FindSet(x):查找元素 x 属于哪个集合，如果 x 属于某一集合，则返回该集合的代表。

UnionSet(x，y):如果元素 x 和元素 y 分别属于不同的集合，则将两个集合合并，否则不做操作。并查集的实现方法是使用有根树来表示集合——树中的每个结点都表示集合的一个元素，每棵树表示一个集合，每棵树的根结点作为该集合的代表。

让我们从一个例题开始了解并查集的操作。

亲戚(relation)

内存限制：256 MiB时间限制：1000 ms标准输入输出

题目类型：传统评测方式：文本比较

题目描述

或许你并不知道，你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱，判断两个人是否是亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。在这种情况下，最好的帮手就是计算机。为了将问题简化，你将得到一些亲戚关系的信息，如Marry和Tom是亲戚，Tom和Ben是亲戚，等等。从这些信息中，你可以推出Marry和Ben是亲戚。请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

输入格式

输入由两部分组成。

第一部分以N，M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1000000)，每行有两个数ai,bi，表示已知ai和bi是亲戚。

第二部分以Q开始。以下Q行有Q个询问(1≤ Q ≤1000000)，每行为ci,di，表示询问ci和di是否为亲戚。

输出格式

对于每个询问ci,di，输出一行：若ci和di为亲戚，则输出“Yes”，否则输出“No”。

样例

样例输入

样例输出

复制Yes
No
Yes

首先，由题目想到此题应是集合应用题，从而想到了并查集。

这里先把并查集的几个函数解释一下。

void makeset () {
    for (int i = 1; i <= n; i ++） {
        fa[i] = i; // 初始化，自己是自己的爸爸
    }
}

int FindSet (int x) { // 找爸爸
	if (fa[x] == x) return x; // 找到了
	else return FindSet (fa[x]); // 没找到，继续找
}

int OnionSet (int x, int y) { // 合并
	fa[FindSet (y)] = FindSet (x); // 建立联系（把y合并到x上，相当于把根接上）
}

这样就可以做到，把所有元素分成若干个亲戚集合。

再用FindSet查询根是否一样，从而判断是否在一个集合。

Tip:优化：路径压缩

对于一个集合中的结点，只需要关心它的根结点是谁，不必知道各结点之间的关系(对树的形态不关心)，希望每个元素到根结点的路径尽可能短，最好只需要一步。把刚才的树转换为右图中的树，极大地提高了查询效率。

路径压缩需要在查询操作时，把沿途的每个结点的父节点都设为根结点即可。下一次再查询时，就可以节约很多时间。

code：

int FindSet (int x) {
	if (fa[x] == x) return x;
	else return fa[x] = FindSet (fa[x]);
}

此题代码：

#include <bits/stdc++.h>
#define sf scanf
#define pf printf
using namespace std;
int fa[50005] = {};
int ran[50005] = {};
int FindSet (int x) {
	if (fa[x] == x) return x;
	else return fa[x] = FindSet (fa[x]);
}
void UnionSet (int x, int y) {
	fa[FindSet (x)] = FindSet (y);
}
int main () {
	int n, m;
	sf ("%d %d", &n, &m);
	for (int i = 1; i <= n; i ++) {
		fa[i] = i;
		ran[i] = 1;
	}
	while (m --) {
		int x, y;
		sf ("%d %d", &x, &y);
		UnionSet (x, y);
	}
	int T;
	sf ("%d", &T);
	while (T --) {
		int x, y;
		sf ("%d %d", &x, &y);
		if (FindSet (x) == FindSet (y)) {
			pf ("Yes\n");
		} else pf ("No\n");
	}
}

其实还有一种优化方式叫按秩合并。

就是把深度低的树往深度高的树上合并，用rank数组记录根结点对应树的深度(如果不是根节点，其rank相当于以它作为根节点的子树的深度)。一开始，把所有元素的rank（秩）设为1。合并时比较两个根结点，把rank较小者往较大者上合并。

但是，压缩路径已经够了，（我也懒得写了）

接下来：

银河英雄传说

内存限制：256 MiB时间限制：1000 ms标准输入输出

题目类型：传统评测方式：文本比较

题目描述

有一个划分为N列的星际战场，各列依次编号为1,2,…,N。

有N艘战舰，也依次编号为1,2,…,N,其中第i号战舰处于第i列。

有T条指令，每条指令格式为以下两种之一：

1、M i j，表示让第i号战舰所在列的全部战舰保持原有顺序，接在第j号战舰所在列的尾部。

2、C i j，表示询问第i号战舰与第j号战舰当前是否处于同一列中，如果在同一列中，它们之间间隔了多少艘战舰。

现在需要你编写一个程序，处理一系列的指令。

输入格式

第一行包含整数T，表示共有T条指令。

接下来T行，每行一个指令，指令有两种形式：M i j或C i j。

其中M和C为大写字母表示指令类型，i和j为整数，表示指令涉及的战舰编号。

输出格式

你的程序应当依次对输入的每一条指令进行分析和处理：

如果是M i j形式，则表示舰队排列发生了变化，你的程序要注意到这一点，但是不要输出任何信息；

如果是C i j形式，你的程序要输出一行，仅包含一个整数，表示在同一列上，第i号战舰与第j号战舰之间布置的战舰数目，如果第i号战舰与第j号战舰当前不在同一列上，则输出-1。

样例

样例输入

复制4
M 2 3
C 1 2
M 2 4
C 4 2

样例输出

复制-1
1

数据范围与提示

N≤30000,T≤500000

按题目描述，此题带了权值

其实也依旧简单，并查集多一个参数d[]

在换列时，要想的一点是：n~k的隔了多少战舰是什么？(可以在文章下面投票，我不说)

我亮一下部分代码:

#include <bits/stdc++.h>
using namespace std;
const int N = 50000;
int fa[N], d[N], SIZE[N];
int find(int x) {
    if (x == fa[x]) {
        return x;
    }
    int root = find(fa[x]);
    d[x] += _______;
    return fa[x] = root;
}
void onion(int x, int y) {
    x = find(x), y = find(y);
    fa[_] = _, d[_] = SIZE[_];
    SIZE[_] += SIZE[_];
}
int main() {
    int T;
    scanf ("%d", &T);
    for (int i = 1; i <= 30000; i++) {
        fa[i] = i;
        SIZE[i] = 1;
	}
    while (T--) {
        char op;
        int x, y;
        scanf("\n%c", &op);
        scanf ("%d%d", &x, &y);
        if (op == 'M') {
            onion(x, y);
        } else {
            if (find(x) == find(y)) {
                printf ("%d\n", ____________);
            } else {
                printf ("-1\n");
            }
        }
    }
}