数据结构——并查集

佛安樊

已于 2022-02-09 23:29:48 修改

阅读量205

点赞数

分类专栏：数据结构文章标签：数据结构 c++ 算法

于 2022-02-06 22:48:26 首次发布

本文链接：https://blog.csdn.net/qq_51539561/article/details/122794302

版权

数据结构专栏收录该内容

1 篇文章 0 订阅

订阅专栏

并查集

路径压缩优化(时间复杂度 $O (1)$ )：

int find(int x)
{
    if (p[x] != x) 
    	p[x] = find(p[x]);//路径压缩优化
    return p[x];
}
int main()
{
//...
		int pa = find(a), pb = find(b);
 		if (*op == 'M') p[pa] = pb;
        else
        {
            if (pa == pb) puts("Yes");
            else puts("No");
        }
        //...
}

size[N]数组表示某点所在的集合的大小，绑定到根节点，只有根节点的size有意义，所以求x所在集合的大小：

//x与y合并
int px = find(x), py = find(y);
if(px != py) 
{
	f[py] = px;//px为根结点
	size[px] += size[py];
}
cout << size[find(x)] << endl;//先确定x的根节点是谁，在size求大小

带权并查集（并查集进阶）

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

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

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

M i j，表示让第 i 号战舰所在列的全部战舰保持原有顺序，接在第 j 号战舰所在列的尾部。
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。

数据范围
N≤30000,T≤500000
在这里插入图片描述
1、size[x]表示集合的大小，d[x]表示x到p[x]的距离

2、将第a列的船接在第b列的末尾，如图所示，相当于让每个点到pb的距离都加上size[pb]，由于存储并查集中存在路径压缩的效果，因此只需要将pa到pb的距离加上size[pb]即可，即d[pa] = size[pb]， 其他跟着pa后面的元素会通过图片2操作对路径进行压缩且更新d[i]的值
（这几句很重要！！）

3、路径压缩的过程

1、找到根结点root

2、计算父节点到上一个父节点的距离d[p[x]],同时进行路径压缩(路径压缩后，上一父节点即root)

3、d[x]初始值是x到p[x]的距离，更新d[x]值，即d[x] = d[x] + d[p[x]]

小呆呆大佬的解释

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 3e4 + 10;

int p[N], d[N], s[N];

int find(int x)
{
    if(p[x] != x)
    {
        int root = find(p[x]);//由于递归的原因，返回到最初的x时,d[p[x]]已经变成p[x]到root的距离
        d[x] = d[x] + d[p[x]];//x的距离进行更新
        p[x] = root;
    }
    return p[x];
}//find函数易写错

int main()
{
    int t;
    cin >> t;
    for(int i = 1; i <= N; i ++ )
    {
        p[i] = i;
        s[i] = 1;
    }
    
    while(t -- )
    {
        char ch[2];
        int a, b;
        scanf("%s%d%d", ch, &a, &b);
        
        int pa = find(a), pb = find(b);
        if(ch[0] == 'M')
        {
            d[pa] = s[pb];
            s[pb] += s[pa];
            p[pa] = pb;
        }
        else
        {
            if(pa != pb)
                cout << -1 << endl;
            else cout << max(0, abs(d[a] - d[b]) - 1) << endl;
        }
    }
    
    return 0;
}