数据结构 - 并查集（合并集合）

1、题目描述

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

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

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

输入格式

第一行输入整数n和m。

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

输出格式

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

每个结果占一行。

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出样例：

Yes
No
Yes

 

2、分析

 

优化：路径压缩

 

3、代码

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {
private static final int N = 100010;
    
    static int[] p = new int[N]; //保存该节点的父节点(也就是上一个结点)
    
    static int find(int x){ //找到x元素的祖宗节点(也就是根节点)
        if(p[x] != x) p[x] = find(p[x]);
        return p[x];
    }
    
    public static void main(String[] args){
        
        Scanner in = new Scanner(new BufferedInputStream(System.in));
        String numsLine = in.nextLine();
        String[] numsArr = numsLine.split(" ");
        int n = Integer.parseInt(numsArr[0]);
        int m = Integer.parseInt(numsArr[1]);
        
        //初始化
        for(int i = 1;i <= n;i ++){
            p[i] = i;
        }
        while(m -- > 0){
            String line = in.nextLine();
            String[] splits = line.split(" ");
            int x = Integer.parseInt(splits[1]);
            int y = Integer.parseInt(splits[2]);
            if(splits[0].equals("M")){
                p[find(x)] = find(y); 
            }else{
                if(find(x) == find(y)){
                    System.out.println("Yes");
                }else{
                    System.out.println("No");
                }
            }
        }
    }
}