洛谷P3367 【模板】并查集

/*
并查集（树实现）
路径压缩、按秩合并
使用前调用init()
find(x) 查找x的根节点(路径压缩)
Union(x,y) 启发式合并 
*/ 
#include<bits/stdc++.h>
using namespace std;
const int size=100005;//集合大小 
int n,m;//n--元素个数 m--查询次数 
struct node{
    int fa,cnt,val;//父节点、自己及子树元素个数（仅当其为根节点始有意义）、元素的值 
}Set[size];
inline int find(int x) {return Set[x].fa==x?x:(Set[x].fa=find(Set[x].fa));}//路径压缩 
inline void Union(int x,int y)
{
    x=find(x),y=find(y);if(x==y) return;
    if(Set[x].cnt>Set[y].cnt) {Set[y].fa=x;Set[x].cnt+=Set[y].cnt;}//按秩合并 
    else {Set[x].fa=y;Set[y].cnt+=Set[x].cnt;}
}
inline void init() {for(int i=1;i<=n;i++) Set[i].fa=Set[i].val=i,Set[i].cnt=1;}//初始化 
int main()
{
    scanf("%d%d",&n,&m);
    init();
    for(int i=1,x,y,z;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(x==1) Union(y,z);
        if(x==2)
        {
            if(find(y)==find(z)) printf("Y\n");
            else printf("N\n");
        }
    }
    return 0;
}