上午写了补题但一直答案错误,在bilibli上看了会有关并查集的视频
下午学长讲课
晚上写并查集的模板的代码
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 N,MN,M ,表示共有 NN 个元素和 MM 个操作。
接下来 MM 行,每行包含三个整数 Z_i,X_i,Y_iZi,Xi,Yi 。
当 Z_i=1Zi=1 时,将 X_iXi 与 Y_iYi 所在的集合合并。
当 Z_i=2Zi=2 时,输出 X_iXi 与 Y_iYi 是否在同一集合内,是的输出 Y
;否则输出 N
。
输出格式
对于每一个 Z_i=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% 的数据,N \le 10N≤10,M \le 20M≤20。
对于 70\%70% 的数据,N \le 100N≤100,M \le 10^3M≤103。
对于 100\%100% 的数据,1\le N \le 10^41≤N≤104,1\le M \le 2\times 10^51≤M≤2×105,1 \le X_i, Y_i \le N1≤Xi,Yi≤N,Z_i \in \{ 1, 2 \}Zi∈{1,2}。
#include<iostream>
using namespace std;
int pre[1000000];
int fd(int i)//寻找根节点
{
if(pre[i]==i)//如果根节点是其本身
return i;
else
{
pre[i]=fd(pre[i]);//直接指向根节点,减小时间复杂度
return pre[i];
}
}
void unn(int i,int j)//合并
{
pre[fd(i)]=fd(j);将i的根节点指向j的根节点
}
void bj(int x,int y)//比较
{
if(fd(x)==fd(y))//如果根节点相同
printf("Y\n");
else
printf("N\n");
}
int main()
{
int n,m;
cin>>n>>m;
int x,y,z;
for(int i=1;i<=n;i++)
pre[i]=i;//所有数的根节点都指向自己
for(int i=1;i<=m;i++){
cin>>z>>x>>y;
if(z==1)
unn(x,y);
else if(z==2)
bj(x,y);
}
//printf("%d %d %d %d",pre[1],pre[2],pre[3],pre[4]);
}