算法-并查集2

在并查集1里面我们学习了并查集的作用，主要实现的功能，并查集2只是对于并查集1的应用的补充。

除了基本实现操作外，并查集还可以考察集合里面点的数量等

题目

题解完整代码（Java）

import java.util.Scanner;

public class Main {
    static int N = 100010;
    static int n;
    static int[] p = new int[N];//存储每个节点的父节点
    static int[] size = new int[N];//每一个集合中点的数量,只有根节点的size[i]才是有意义的，合并时点的数量才可以进行运算

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        int m = sc.nextInt();
        init();
        while (m-- > 0) {
            String s = sc.next();
            if (s.equals("C")) {
                int a = sc.nextInt();
                int b = sc.nextInt();
                if (find(a) == find(b)) {
                    continue;//如果两个数是同一集合中的数，就结束
                } else {
                    size[find(b)] += size[find(a)];//由于是把a集合并入b集合中，所以把a集合的点的个数加到b的点的个数上
                    p[find(a)] = find(b);//合并操作：让a的祖宗节点的父节点指向b的祖宗节点 ——> a集合并入b集合中
                }
            } else if (s.equals("Q1")) {
                int a = sc.nextInt();
                int b = sc.nextInt();
                if (find(a) == find(b)) {//查询a、b是否在同一集合内：如果a和b的祖宗节点是同一个的话，证明是同一个集合，否则，就不属于同一个集合
                    System.out.println("Yes");
                } else {
                    System.out.println("No");
                }
            } else {
                int a = sc.nextInt();
                System.out.println(size[find(a)]);
            }
        }
    }

    //初始化，一开始所有节点的父节点都指向自己，每一个集合中只有自己这个点，所以size[i]初始化为1
    public static void init() {
        for (int i = 1; i <= n; i++) {
            p[i] = i;
            size[i] = 1;
        }
    }

    //返回x的祖宗节点 + 路径压缩
    public static int find(int x) {
        if (p[x] != x) {//如果x的父节点不是它本身的话，证明x不是祖宗节点，就让x的父节点接着找父节点，直到找到祖宗节点。并且让x的父节点指向祖宗节点,返回祖宗节点。
            p[x] = find(p[x]);
        }
        return p[x];
    }
}

详解：

和之前实现基本功能的代码的区别是：
多了size[]数组记录每个集合中点的数量
需要注意的是：
由于两集合合并，有可能给的是同一个集合，这就需要我们进行判断，判断两个点是否属于同一集合：

if (find(a) == find(b)) {
                    continue;//如果两个数是同一集合中的数，就结束
                } else {
                    size[find(b)] += size[find(a)];//由于是把a集合并入b集合中，所以把a集合的点的个数加到b的点的个数上
                    p[find(a)] = find(b);//合并操作：让a的祖宗节点的父节点指向b的祖宗节点 ——> a集合并入b集合中
                }

并查集2主要是关于并查集1的基本功能的额外应用，之后写的并查集题目也会进行补充。