洛谷P3367【模板】并查集

洛谷P3367 【模板】并查集

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

第一行包含两个整数 $N$,$M$ ,表示共有 $N$ 个元素和 $M$ 个操作。

接下来 $M$ 行，每行包含三个整数 $Z_i$,$Xi$,$Yi$。

当 $Z_i$=1 时，将$Xi$与 $Yi$ 所在的集合合并。

当 $Z_i$=2 时，输出 $Xi$ 与 $Yi$ 是否在同一集合内，是的输出 $Y$ ；否则输出 $N$ 。

输出格式

对于每一个 $Z_i$=2 的操作，都有一行输出，每行包含一个大写字母，为 $Y$ 或者。$N$

输入输出样例

输入 #1

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出 #1

N
Y
N
Y

说明/提示

对于 $30$ $\%$的数据，$N$$\le$$10$，$M\le 20$ 。

对于 $70$ $\%$的数据，$N\le 100$，$M\le 10^3$ 。

对于 $100$ $\%$的数据，$1\le N\le 10^4$，$1\le M\le 2\times 10^5$ 。

代码实现

---

import java.util.Scanner;

public class P3367 {

    public static int n;        //元素个数

    public static void main(String[] args) {

        Scanner sc=new Scanner(System.in);
        n = sc.nextInt();
        int m = sc.nextInt();

        int[] parent = new int[n+1];
        int[] rank = new int[n+1];
        init(parent,rank);

        for(int i=0;i<m;i++){

            int z = sc.nextInt();
            int x = sc.nextInt();
            int y = sc.nextInt();

            if(z == 1){
                //连接元素
                union_root(x,y,parent,rank);
            }else if(z == 2){
                //查看元素是否在同一个集合中
                int result = equ_root(x,y,parent);
                if(result == 0){
                    System.out.println("Y");
                }else{
                    System.out.println("N");
                }
            }
        }
    }

    public static void init(int parent[], int rank[]){
        for(int i=1;i<=n;i++){     //初始化数组
            parent[i] = -1;
            rank[i] = 0;
        }
    }

    public static int find_root(int x ,int parent[]){
        int x_root = x;                 //查找根节点
        while(parent[x_root] != -1){
            x_root = parent[x_root];
        }
        return x_root;
    }

    /* 
       判断两个元素是否在同一集合中
       0-在同一个集合中， 1-不在同一个集合中
    */
    public static int equ_root(int x, int y, int parent[]){
        int x_root = find_root(x,parent);      
        int y_root = find_root(y,parent);
        if(x_root == y_root){
            return 0;
        }else{
            return 1;
        }
    }
    
    /* 合并元素 */
    public static void union_root(int x, int y, int parent[], int rank[]){
        int x_root = find_root(x,parent);
        int y_root = find_root(y,parent);

        if(x_root != y_root){
            if(rank[x_root] > rank[y_root]){
                parent[y_root] = x_root;
            }else if(rank[x_root] < rank[y_root]){
                parent[x_root] = y_root;
            }else{
                parent[x_root] = y_root;
                rank[y_root]++;
            }
        }
    }
}