Codeforces::GYM 102896A Almost Balanced Tree

题目链接

题目大意

定义“几乎平衡的树”（后文简称ABT）为：

• 它是一棵二叉树
• 对于任意节点，左子树的权值和以及右子树的权值和，相差不超过 $1$

现在有 $a$ 个 $1$ 和 $b$ 个 $2$，作为各个点的权值。要求构造一棵ABT，输出结果，若不存在输出 $-1$

解题思路

我们发现，权值为 $2$ 的点是很令人讨厌的，因为如果不能构成 ABT，一定是因为奇数个 $2$ 在没有 $1$ 的情况下无法平均分配给两棵子树。而且，如果某个 $a,b$ 能构造出ABT，把其中一个 $2$ 替换成 $1$ 一定是有解的，但把 $1$ 替换成 $2$ 就不一定行。
我们不妨贪心构造。由于 $2$ 容易造成失衡，把 $2$ 放在根部是一种比较合适的方法，这样可以减少子树中 $2$ 的个数，而且可以省下一个 $1$ （当然如果没有 $2$ ，就只能把 $1$ 放在根部了）。我们剩余了若干个 $2$ 和 $1$，为了让 $2$ 最多的子树中的 $2$ 的数量尽可能少，我们平均分配 $2$。剩下的 $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;
}