题目大意
定义“几乎平衡的树”(后文简称ABT)为:
- 它是一棵二叉树
- 对于任意节点,左子树的权值和以及右子树的权值和,相差不超过 1 1 1
现在有 a a a 个 1 1 1 和 b b b 个 2 2 2,作为各个点的权值。要求构造一棵ABT,输出结果,若不存在输出 − 1 -1 −1
解题思路
我们发现,权值为
2
2
2 的点是很令人讨厌的,因为如果不能构成 ABT,一定是因为奇数个
2
2
2 在没有
1
1
1 的情况下无法平均分配给两棵子树。而且,如果某个
a
,
b
a,b
a,b 能构造出ABT,把其中一个
2
2
2 替换成
1
1
1 一定是有解的,但把
1
1
1 替换成
2
2
2 就不一定行。
我们不妨贪心构造。由于
2
2
2 容易造成失衡,把
2
2
2 放在根部是一种比较合适的方法,这样可以减少子树中
2
2
2 的个数,而且可以省下一个
1
1
1 (当然如果没有
2
2
2 ,就只能把
1
1
1 放在根部了)。我们剩余了若干个
2
2
2 和
1
1
1,为了让
2
2
2 最多的子树中的
2
2
2 的数量尽可能少,我们平均分配
2
2
2。剩下的
1
1
1 怎么分配呢?其实只有一种方法能让左右权值差不超过
1
1
1 了,因为总的权值和是不变的。递归下去建树即可,如果到了某一步玩不下去了,那就输出
−
1
-1
−1 退出即可
参考代码
#include <bits/stdc++.h>
using namespace std;
int l[100010], r[100010], w[100010];
int cnt = 0;
void build(int a, int b, int num){
if(b > 0){
b --;
w[num] = 2;
}else{
a --;
w[num] = 1;
}
int lb = b / 2;
int rb = b - b / 2;
int sum = a + b * 2;
int la = sum / 2 - lb * 2;
int ra = a - la;
if(la < 0 or ra < 0){
cout << -1;
exit(0);
}
if(la or lb){
l[num] = ++ cnt;
build(la, lb, cnt);
}
if(ra or rb){
r[num] = ++ cnt;
build(ra, rb, cnt);
}
}
int main(){
int a, b;
cin >> a >> b;
build(a, b, ++ cnt);
for(int i = 1; i <= a + b; i ++){
cout << w[i] << ' ' << l[i] << ' ' << r[i] << '\n';
}
return 0;
}