【洛谷刷题】P3367 【模板】并查集

题目链接：P3367 【模板】并查集 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题目描述

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

输入格式

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

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

当 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

说明/提示

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

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

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

解题代码 ：

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5 + 2;
int boss[MAXN];
int n;
void init(int n)
{
	for (int i = 1; i <=n; i++)
	{
		boss[i] = i;
	}
}
int findb(int x)
{
	if (x == boss[x])
	{
		return x;
	}
	else
	{
		int max_boss = findb(boss[x]);
		return boss[x] = max_boss;
	}
}
void merge(int x, int y)
{
	int px = findb(x);
	int py = findb(y);
	if (px != py)
	{
		boss[px] = py;              //这个地方出了问题，应该是boss[px]=py;直接将最大的老板收入集团   
	}
}

bool judge(int x1, int x2)
{
	return findb(x1) == findb(x2);
}


int main()
{
	cin >> n;
	init(n);
	int m;
	cin >> m;
	for (int i = 0; i < m; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;
		if (a == 1) {
			merge(b, c);
		}
		else
		{
			if (judge(b, c))
			{
				cout << "Y" << endl;
			}
			else
			{
				cout << "N" << endl;
			}
		}
	}
	return 0;
}

错的点：

（1）最开始init()函数没有调用

（2）find()函数里面，重新赋值时，要将x的boss的boss重新赋值，为boss[boss[x]]=boss[y]！！