【题解-洛谷】P3367 【模板】并查集

题目：P3367 【模板】并查集

题目背景

本题数据范围已经更新到 $1\le N\le 2\times 10^5$，$1\le M\le 10^6$。

题目描述

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

输入格式

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

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

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

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

输出格式

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

输入输出样例 #1

输入 #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

说明/提示

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

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

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

对于 $100\%$ 的数据，$1\le N\le 2\times 10^5$，$1\le M\le 10^6$，$1\le X_i,Y_i\le N$， Z i ∈ { 1 , 2 } Z_i \in \{ 1, 2 \} Zi​∈{1,2}。

代码

#include<iostream>

using namespace std;

const int MaxN = 2e5 + 10;

int N, M;
int p[MaxN];

void init(){
    for(int i = 1; i <= N; i ++){
        p[i] = i;
    }
}

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

void merge(int x, int y){
    int px = findd(x), py = findd(y);
    if(px != py){
        p[px] = py;
    }
}

bool query(int x, int y){
    return findd(x) == findd(y);
}

int main(){
    scanf("%d%d", &N, &M);
    init();
    while(M --){
        int op, x, y;
        scanf("%d%d%d", &op, &x, &y);
        if(op == 1){
            merge(x, y);
        }
        else{
            if(query(x, y)){
                puts("Y");
            }
            else{
                puts("N");
            }
        }
    }
    return 0;
}

结果