P3367 【模板】并查集

题目

题目描述

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

输入格式

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

接下来 �M 行，每行包含三个整数 ��,��,��Zi,Xi,Yi 。

当 ��=1Zi=1 时，将 ��Xi 与 ��Yi 所在的集合合并。

当 ��=2Zi=2 时，输出 ��Xi 与 ��Yi 是否在同一集合内，是的输出 Y ；否则输出 N 。

输出格式

对于每一个 ��=2Zi=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

说明/提示

对于 30%30% 的数据，�≤10N≤10，�≤20M≤20。

对于 70%70% 的数据，�≤100N≤100，�≤103M≤103。

对于 100%100% 的数据，1≤�≤1041≤N≤104，1≤�≤2×1051≤M≤2×105，1≤��,��≤�1≤Xi,Yi≤N，��∈{1,2}Zi∈{1,2}。

//以下为错误代码 自留检查

情况：输入都一个x y z程序结束

原因：find函数写错了（ku）

勤加练习

//P3367 【模板】并查集
#include<bits/stdc++.h>
using namespace std;
#define MAXN 100005
int fa[MAXN];
void init(int n)
{
    for(int i=1;i<=n;i++)
      fa[i]=i;
}
int find(int x)
{
    if(x==find(x))
       return x;
    else{
        fa[x]=find(fa[x]);
        
        return fa[x];
    }
}
void unionn(int i,int j)
{
    int i_fa=find(i);
    int j_fa=find(j);
    fa[i_fa]=j_fa;
}
int main()
{
    int n,m;
    int z,x,y;
    cin>>n>>m;
    init(n);
    while(m--)
    {
        scanf("%d%d%d",&x,&y,&z);
        switch(z)
        {
            case 1:unionn(x,y); break;
            case 2:
                if(find(x)==find(y)) cout<<"Y"<<endl;break;
                cout<<"N"<<endl;
        }
    }
}

正确：

//P3367 【模板】并查集
#include<bits/stdc++.h>
using namespace std;
#define MAXN 100005
int fa[MAXN];
void init(int n)
{
    for(int i=1;i<=n;i++)
      fa[i]=i;
}
int find(int x)
{
    if(x==fa[x])
       return x;
    else{
        fa[x]=find(fa[x]);
        
        return fa[x];
    }
}
void unionn(int i,int j)
{
    int i_fa=find(i);
    int j_fa=find(j);
    fa[i_fa]=j_fa;
}
int main()
{
    int n,m;
    int z,x,y;
    cin>>n>>m;
    init(n);
    while(m--)
    {
        scanf("%d%d%d",&z,&x,&y);
        if(z==1)
        {
            unionn(x,y);
        }
        else if(z==2)
        {
            if(find(x)==find(y))
               cout<<"Y"<<endl;
            else
               cout<<"N"<<endl;
        }
        else
        return 0;
    }
    return 0;
}
 

i like ac