并查集——竞赛应用小结_并查集find(a)==find(b)-CSDN博客

本文链接：https://blog.csdn.net/ChuRi_BaiYu/article/details/126154575

一、概念

概念来源：OI Wiki

并查集：并查集是一种树形的数据结构，顾名思义，它用于处理一些不交集的合并及查询问题。

二、图示说明

集合1,2,3,4,5,有一个相同的祖宗节点1，而集合6,7,8有一个相同的祖宗节点6。且集合1也还是集合2，3的父节点。

三、操作说明

1.集合合并

如图2，图3所示，初始时集合1,2,3的父结点和祖宗结点都是自己，要合并集合1，2。将集合1和集合2合并在一起，让集合2的祖宗结点作为集合1的的父节点即可

int p[N]; ///存储每一个集合的父结点
int n;

for(int i=1;i<=n;i++)   p[i]=i; ///初始时，每个集合的父节点为自己

int find(int x){
    ///当x集合的父节点不为自己时，也说明x集合不是祖宗结点，让p[x]=find(p[x]),以p[x]集合作为find()函数参数，递归找到x集合的祖宗结点，并让其作为x集合的父节点
    if(x!=p[x]) p[x]=find(p[x]); 
    return p[x]; ///返回祖宗节点
}

p[find(a)]=find(b); ///让b集合的祖宗节点作为a集合祖宗结点的父节点

2.集合查询


if(find(a)==find(b)) ///当集合a,b的祖宗节点相同时，说明这两个集合在一个集合内中
{
    ///某种操作；
}

3.集合数量累加

如图4,5,6，将集合1,2合并。先将在集合1的祖宗结点中存储的集合总个数累加到集合2的祖宗结点上，然后再让集合2的祖宗结点成为集合1祖宗结点的父结点

int p[N];
int si[N]; ///存储集合所在的集合内的集合个数 

for (int i = 1; i <= n; i++){
        p[i] = i;
        si[i] = 1; ///初始时，每个集合中只有集合本身这一个集合
}

cin >>a >>b;
if (find(a) != find(b)){
    ///当两个集合的祖宗节点不同时，先让a集合祖宗结点中存储的集合内总数累加到b集合的祖宗结点中，然后再让a集合的祖宗结点作为b集合祖宗结点的子节点
    si[find(b)] += si[find(a)]; 
    p[find(a)] = find(b);
}

4.集合数量查询

 cout << si[find(a)] << endl; ///输出集合a所在集合的集合个数

5.带权集合合并

如图7,8,9所示，让集合7的祖宗结点成为集合2祖宗结点的子节点。先让集合7祖宗结点成为集合7的父节点，同时更新权值，然后再让集合7的祖宗结点成为集合2祖宗结点的子节点，同时更新权值。

int p[N];
int d[N]; ///每一个集合到父节点的权值
int n;
int find(int x) {
    if (x != p[x]) {
        int t=find(p[x]); ///让t等于x集合的祖宗结点
        d[x] +=d[p[x]]; ///让x集合到父节点的权值变为成x集合到祖宗结点的权值，等价于x集合到父节点的权值加上x集合父节点到祖宗结点的权值
        p[x] = t; ///最后让x集合指向其祖宗结点
    }
    return p[x];
}

for (int i = 1; i <= n; i++) {
    p[i] = i;
    d[i] = 0; ///初始时，每个集合到其祖宗结点的权值为零
}

cin >>x >>y;
p[find(x)] = find(y); ///让x集合祖宗节点成为y集合祖宗节点的子节点
d[find(x)] = d[y] - d[x]+z; ///x集合祖宗结点到y集合祖宗节点的权值，等于y集合到祖宗结点的权值减去x集合到其祖宗结点的权值在加上偏量z（偏量z，按具体情况来看）

6.带权集合查询

cin >>a;
cout <<d[a] <<endl;

四、例题实践

1.简单并查集

a.题目描述

题目来源：acwing

一共有 n 个数，编号是 1∼n，最开始每个数各自在一个集合中。

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

M a b，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；

输入格式

第一行输入整数 n 和 m。

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

输出格式

对于每个询问指令 Q a b，都要输出一个结果，如果 a 和 b 在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

数据范围

1≤n,m≤1e5

输入样例：

输出样例：

Yes
No
Yes

b.解题思路

思路：按题目要求，使用并查集解决

c.代码实现

#include<iostream>

using namespace std;

const int N=1e5+10;

int p[N]; ///存储每一个集合的父结点
int n,m;

int find(int x){
    ///当x集合的父节点不为自己时，也说明x集合不是祖宗结点，让p[x]=find(p[x]),以p[x]集合作为find()函数参数，递归找到x集合的祖宗结点，并让其作为x集合的父节点
    if(x!=p[x]) p[x]=find(p[x]); 
    return p[x]; ///返回祖宗节点
}

int main(){
    cin >>n >>m;
    for(int i=1;i<=n;i++)   p[i]=i; ///初始时，每个集合的父节点为自己
    while(m--){
        char op;
        cin >>op;
        int a,b;
        if(op=='M'){
            cin >>a >>b;
            p[find(a)]=find(b); ///让b集合的祖宗节点作为a集合祖宗结点的父节点
        }
        else{
            cin >>a >>b;
            if(find(a)==find(b))    puts("Yes"); ///当集合a,b的父节点相等时，说明这两个集合在一个集合中
            else puts("No");
        }
    }
    return 0;
}

2.带数量并查集

a.题目描述

题目来源：acwing

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

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

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

输入格式

第一行输入整数 n和 m。

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

输出格式

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

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

每个结果占一行。

数据范围

1≤n,m≤1e5

输入样例：

输出样例：

Yes
2
3

b.解题思路

思路：使用带数量的并查集，按题目要求来解决问题

c.代码实现

#include<iostream>

using namespace std;

const int N = 1e5 + 10;

int p[N];
int si[N]; ///存储集合所在的集合内的集合个数 
int m, n;

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

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++){
        p[i] = i;
        si[i] = 1; ///初始时，每个集合中只有集合本身这一个集合
    }

    while (m--) {
        char op[2];
        int a, b;
        cin >> op;
        if (op[0] == 'C') {
            cin >> a >> b;
            if (find(a) == find(b))    continue;
            ///当两个集合的祖宗节点不同时，先让a集合祖宗结点中存储的集合内总数累加到b集合的祖宗结点中，然后再让a集合的祖宗结点作为b集合祖宗结点的子节点
            si[find(b)] += si[find(a)]; 
            p[find(a)] = find(b);
        }
        else if (op[1]== '1') {
            cin >> a >> b;
            if (find(a) == find(b))    puts("Yes");
            else puts("No");
        }
        else {
            cin >> a;
            cout << si[find(a)] << endl; ///输出集合a所在集合的集合个数
        }
    }
}

3.带权并查集

a.题目描述

题目来源：acwing

动物王国中有三类动物 A,B,C，这三类动物的食物链构成了有趣的环形。

A 吃 B，B 吃 C，C 吃 A。

现有 N 个动物，以 1∼N 编号。

每个动物都是 A,B,C 中的一种，但是我们并不知道它到底是哪一种。

有人用两种说法对这 N 个动物所构成的食物链关系进行描述：

第一种说法是 1 X Y，表示 X 和 Y 是同类。

第二种说法是 2 X Y，表示 X 吃 Y。

此人对 N 个动物，用上述两种说法，一句接一句地说出 K 句话，这 K 句话有的是真的，有的是假的。

当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

当前的话与前面的某些真的话冲突，就是假话；
当前的话中 X 或 Y 比 N 大，就是假话；
当前的话表示 X 吃 X，就是假话。

你的任务是根据给定的 N 和 K 句话，输出假话的总数。

输入格式

第一行是两个整数 N 和 K，以一个空格分隔。

以下 KK 行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中 D 表示说法的种类。

若 D=1，则表示 X 和 Y 是同类。

若 D=2，则表示 X 吃 Y。

输出格式

只有一个整数，表示假话的数目。

数据范围

1≤N≤50000,
0≤K≤100000

输入样例：

输出样例：

b.解题思路

思路：在三种动物中建立权值关系，权值模3可得0,1,2三种关系，将这三种关系与三种动物联系起来，同种动物的权值关系相同，不同种动物关系，遵循2吃1，1吃0,0吃1。以此建立一个带权值的并查集。

c.代码实现

#include<iostream>

using namespace std;

const int N = 5e4;
int p[N];
int d[N]; ///每一个集合到父节点的权值
int n, m;

int find(int x) {
    if (x != p[x]) {
        int t=find(p[x]); ///让t等于x集合的祖宗结点
        d[x] +=d[p[x]]; ///让x集合到父节点的权值变为成x集合到祖宗结点的权值，等价于x集合到父节点的权值加上x集合父节点到祖宗结点的权值
        p[x] = t; ///最后让x集合指向其祖宗结点
    }
    return p[x];
}

int main() {
    cin >> n >> m;
    int res = 0; ///统计假话数量
    for (int i = 1; i <= n; i++) {
        p[i] = i;
        d[i] = 0; ///初始时，每个集合到其祖宗结点的权值为零
    }
    while (m--) {
        int dt, x, y;
        cin >> dt >> x >> y;
        if (x > n || y > n) { ///当x或y大于动物的数量时为假
            res++;
            continue;
        }
        if (dt == 1) {
            int fx = find(x), fy = find(y);
            if (fx == fy) {
                ///若x集合到其祖宗结点的权值关系不等于y集合到祖宗结点的权值权值关系，即权值模3后不相等为假
                if ((d[x] - d[y]) % 3 != 0)  { 
                    res++;
                }
            }
            else {
                p[fx] = fy;///让x集合祖宗节点成为y集合祖宗节点的子节点
                ///x集合到其祖宗结点的权值再加上x集合祖宗结点到y集合祖宗节点的权值，等于y集合到祖宗结点的权值
                d[fx] = d[y] - d[x];
            }
        }
        else {
            if (x == y) { ///同类不能吃同类
                res++;
                continue;
            }
            int fx = find(x), fy = find(y);
            if (fx == fy) {
                 ///若x集合到其祖宗结点的权值减1关系不等于y集合到祖宗结点的权值权值关系，即x权值-1模3后不等于y权值模3为假
                if ((d[x]-d[y]-1)%3!=0)  {
                    res++;
                }
            }
            else {
                p[fx] = fy;
                ///x集合到其祖宗结点的权值再加上x集合祖宗结点到y集合祖宗节点的权值减1，等于y集合到祖宗结点的权值
                d[fx] = d[y] - d[x] + 1;
            }
        }
    }
    cout << res << endl; ///假话数量
    return 0;
}