数据结构——(计数)并查集——连通块中点的数量

题目描述

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

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

1.C a b，在点 a 和点 b 之间连一条边，a 和 b 可能相等；
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 所在连通块中点的数量

每个结果占一行。

数据范围
1≤n,m≤10⁵
输入样例：

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

输出样例：

Yes
2
3

题目分析

1.连边操作可以看做并集操作
2. 询问是否在一个块中，可以看成查集操作
3. 在并查集的基础上进行计数操作：初始每个集合中的元素个数都是1，用size[]数组来维护数量，size数组存的是根节点为x的元素的个数即一个集合当中的元素的个数，只需在并集操作的时候，看是否在一个集合当中，如果在则不用累加直接跳过，如果不在则需要计数，size[find(y)]+=size[find(x)];
4. 特别注意：
size只有祖节点的有意义
要特别注意所有处理size的地方，都要“归根结底”

解决方案

import java.util.*;

public class Main{
    static int N = 100010,n,m;
    static int[] p = new int[N];
    static int[] size = new int[N];
    
    static int find(int x){
        if(p[x]!=x)p[x] = find(p[x]);
        return p[x];
    }
    
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        String[] s1 = sc.nextLine().split(" ");
        n = Integer.parseInt(s1[0]);
        m = Integer.parseInt(s1[1]);
        //初始化
        for(int i=0;i<n;i++){
            p[i] = i;
            size[i] = 1;
        }
        
        while(m-->0){
            String[] s2 = sc.nextLine().split(" ");
            String op = s2[0];
            
            if(op.equals("C")){
                int x = Integer.parseInt(s2[1]);
                int y = Integer.parseInt(s2[2]);
                if(find(x)==find(y)) continue;
                size[find(y)]+=size[find(x)];
                p[find(x)] = find(y);
            }else if(op.equals("Q1")){
                int x = Integer.parseInt(s2[1]);
                int y = Integer.parseInt(s2[2]);
                if(find(x)==find(y)) System.out.println("Yes");
                else System.out.println("No");
            }else{
                int x = Integer.parseInt(s2[1]);
                System.out.println(size[find(x)]);
            }
        }
        
    }
    
}