AtCoder Beginner Contest 279-F BOX（补题）「并查集」

原题链接F - BOX (atcoder.jp)

思路： 

我们让每次第一个进盒的球作为盒最大的儿子（可以想成小组长），要走的时候直接把这一支接到另一支上去。其中并查集只维护球，找出来的也是这个盒最大的儿子，只需输出这个儿子所属的盒即可。

AC代码 

#include <bits/stdc++.h>
using namespace std;
#define io ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
const int N = 6e5 + 8;
int f[N], s[N], he[N];
int n, q, k;

void init() {
	for (int i = 1; i <= n; ++i) {
		f[i] = i; //每个元素的爸爸
		s[i] = i; //盒的特殊元素
		he[i] = i; //特殊元素所在的盒子
	}
}

int find(int a) {
	if (f[a] == a)
		return a;
	else
		return f[a] = find(f[a]);
}

void work() {
	int op, x, y;
	cin >> op >> x;
	switch (op) {
		case 1:
			cin >> y;
			if (s[x] == 0) {
				s[x] = s[y];
				he[s[y]] = x;//盒空就直接让小组长换爸爸
			} else
				f[s[y]] = s[x], he[s[y]] = 0;//不空就让小组长降辈成为原组长的儿子
			s[y] = 0;
			break;
		case 2:
			if (s[x] != 0)
				f[++k] = s[x];//盒不空就让他当小儿子
			else
				he[++k] = x, s[x] = k, f[k] = k;//空就让他当大儿子，记得初始化让他的爸爸等于他
			//cout << "k" << k << "x" << x << "s" << s[x] << '\n';
			//cout << k << '\n';
			break;
		case 3:
			cout << he[find(x)] << '\n';
			break;
	}
}

int main() {
	io;
	cin >> n >> q;
	k = n;
	init();
	while (q--) {
		work();
	}
	return 0;
}

 另：

我一开始写的是只开了两个数组，让f既维护球与球的关系又维护球与盒的关系，这样找爹函数我想的是，当他爸爸的儿子是他的时候，说明找到底了。但是没找到底的情况不会写了，现在这个递归修改了s数组，这是不应该的。师哥说还有可能出现球和盒下标重合的情况（没手推例子，但是感觉会有点问题），如果有什么好想法希望可以指点一下。

错误代码

#include <bits/stdc++.h>
using namespace std;
#define io ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
const int N = 6e5 + 8;
int f[N], s[N];
int n, q, k;

void init() {
	for (int i = 1; i <= n; ++i) {
		f[i] = i; //每个元素的爸爸
		s[i] = i; //盒的特殊元素
	}
}

int find(int a) {
	if (s[f[a]] == a)
		return f[a];
	else
		return s[f[a]] = find(f[a]) ;
}
//这个递归是不对的

void work() {
	int op, x, y;
	cin >> op >> x;
	switch (op) {
		case 1:
			cin >> y;
			if (s[x] == 0) {
				s[x] = s[y];
				f[s[y]] = x;
			} else
				f[s[y]] = s[x];
			s[y] = 0;
			break;
		case 2:
			if (s[x] != 0)
				f[++k] = s[x];
			else
				f[++k] = x, s[x] = k;
			//cout << "k" << k << "x" << x << "s" << s[x] << '\n';
			//cout << k << '\n';
			break;
		case 3:
			cout << find(x) << '\n';
			break;
	}
}

int main() {
	io;
	cin >> n >> q;
	k = n;
	init();
	while (q--) {
		work();
	}
	return 0;
}

 谢谢yjgg不厌其烦的教导。