C++并查集模板

说明：

学习该模板比较推荐洛谷P3367 【模板】并查集，以下是该题链接及描述：
https://www.luogu.org/problemnew/show/P3367

题目描述

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

输入格式：

第一行包含两个整数N、M，表示共有N个元素和M个操作。
接下来M行，每行包含三个整数Zi、Xi、Yi
当Zi=1时，将Xi与Yi所在的集合合并
当Zi=2时，输出Xi与Yi是否在同一集合内，是的话输出Y；否则话输出N

输出格式：

如上，对于每一个Zi=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

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据，N<=10，M<=20；
对于70%的数据，N<=100，M<=1000；
对于100%的数据，N<=10000，M<=200000。

提示：

要抓住每个元素的父亲，合理应用路径压缩：查找完根节点后，在递归回来时顺便将路径上的元素的父亲指向根节点。代码不长的。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#define N 1000001
using namespace std;
int f[N],n,m;
int find(int k)
{
    if(k!=f[k]) f[k]=find(f[k]);
    return f[k];
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&z,&x,&y);
        x=find(x);
        y=find(y);
        if(z==1&&x!=y) f[x]=y;
        if(z==2)
        {
            if(x==y) printf("Y\n");
            else printf("N\n");
        }
    }
    return 0;
}