CodeForce[1500-2000]——1950F - 0, 1, 2, Tree!

codeforces刷题日记

题目大意：要求你构造一棵树，树有三种节点，a个节点有两颗子树，b个节点有一颗子树，c给节点是叶子节点，求这样一颗树的最小深度。

思路：首先可以确定这是一颗二叉树，那我们可以先利用二叉树性质：a=c+1（可以想象一下，每次为一个叶子节点添加两条边，叶子节点+1，度为2的节点也+1；而为叶子节点只加一条边时，叶子节点数量不变），这样就可以判断题目给的节点数是否合理，不合理直接输出-1；只要合法了，那我们就不用管c了，因为根据a构建出来的二叉树一定c=a+1，我们要做的只要添加度为1和2的节点就可以，那如何让深度最小呢，那我们肯定要贪心的策略，尽可能先放度为2的，这样在一层的节点可以最多，在实现的时候，我们不需要创建一颗树，我们用curr和next保存当前深度和下一深度的节点树，然后迭代遍历即可。

#include<iostream>
using namespace std;
int t,a,b,c;
void solve() {
	int a, b, c;
	cin >> a >> b >> c;
	if (a + 1 != c) {cout << -1 << '\n'; return;}
	if (a + b + c == 1) {cout << 0 << '\n'; return;}
	int curr = 1, next = 0, res = 1;
	for (int i = 0; i < a + b; i++) {
		if (!curr) {
	        swap(next, curr);
	        res++;
	    }
	    curr--;
	    next++;
	    if (i < a) {next++;}
	}
	cout << res << '\n';
}
int main(){
    cin>>t;
    while(t--){
        solve();
    }
    return 0;
}