【模板】并查集（洛谷P3367）

Description

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

Input

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

Output

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

Solution

路径压缩

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,f[10010],opt,x,y;
inline int read()
{
    int ans=0;
    char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9')
    {
        ans=ans*10+ch-'0';
        ch=getchar();
    }
    return ans;
}
int find(int x)
{
    if (f[x]==x) return x;
    return f[x]=find(f[x]);
}
int main()
{
    n=read(),m=read();
    for (int i=1;i<=n;i++) f[i]=i;
    while (m--)
    {
        opt=read(),x=read(),y=read();
        if (opt==1)
        {
            int fx=find(x),fy=find(y);
            if (fx!=fy) f[fx]=fy;
        }
        else
        {
            int fx=find(x),fy=find(y);
            if (fx!=fy) printf("N\n"); else printf("Y\n");
        }
    }
    return 0;
}

按秩合并

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,f[100010],dep[100010],opt,x,y;
inline int read()
{
    int ans=0;
    char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9')
    {
        ans=ans*10+ch-'0';
        ch=getchar();
    }
    return ans;
}
int find(int x)
{
    if (f[x]==x) return x;
    return find(f[x]);
}
int main()
{
    n=read(),m=read();
    for (int i=1;i<=n;i++) f[i]=i,dep[i]=0;
    while (m--)
    {
        opt=read(),x=read(),y=read();
        if (opt==1)
        {
            int fx=find(x),fy=find(y);
            if (fx==fy) continue;
            if (dep[fx]<dep[fy]) f[fx]=fy;
            else if (dep[fx]>dep[fy]) f[fy]=fx;
            else
            {
                f[fx]=fy;
                dep[fy]++;
            }
        }
        else
        {
            int fx=find(x),fy=find(y);
            if (fx!=fy) printf("N\n"); else printf("Y\n");
        }
    }
    return 0;
}