并查集【模板】

7-13 并查集【模板】 （30 分）

给出一个并查集，请完成合并和查询操作。

输入格式:
第一行包含两个整数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。

输入样例:

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

输出样例:

N
Y
N
Y

数据规模:

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

上代码：

#include <bits/stdc++.h>
using namespace std;

#define MAX 10000
int Tree[MAX];

void setTree()
{
    for(int i=0; i<MAX; i++)
    {
        Tree[i]=i;
    }
}

int findRoot(int x)
{
    if(Tree[x]==x)return x;
    int tem=findRoot(Tree[x]);
    Tree[x]=tem;
    return tem;
}

void union_set(int a,int b)
{
    int fa=findRoot(a);
    int fb=findRoot(b);
    if(fa!=fb)Tree[fa]=fb;
}

int main()
{
    int N,plan;
    scanf("%d%d",&N,&plan);
    setTree();
    for(int i=0; i<plan; i++)
    {
        int select;
        scanf("%d",&select);
        if(select==1)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            union_set(a,b);
        }

        if(select==2)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(findRoot(a)!=findRoot(b))
            {
                printf("N\n");
            }
            else
            {
                printf("Y\n");
            }
        }

    }
    return 0;
}