HNU实验三 树 编程题 1. 0-1-2树 （残次版）

【问题描述】

    找出具有a+b+c个节点且高度最小的二叉树，满足以下条件：               

• a个节点具有2个子节点

• b个节点具有1个子节点

• c个节点没有子节点

如果不存在这样的树，则输出-1， 否则输出树的高度。树的高度是指从根节点到最底层叶节点之间的边数之和。

【输入形式】

  输入的第一行为T，表示有T个测试样例，接下来的T行，每行3个整数a、b、c(0≤a、b、c≤105, a+b+c≥1)，分别表示所构成的树有a+b+c个节点，其中a个节点具有2个子节点，b个节点具有1个子节点，c个节点没有子节点。
【输出形式】

  输出为T行，每行一个整数，表示满足条件的树的最小高度，如果不存在这样的树，则输出-1。
【样例输入】

10
2 1 3
0 0 1
0 1 1
1 0 2
1 1 3
3 1 4
8 17 9
24 36 48
1 0 0
0 3 1

【样例输出】

2
0
1
1
-1
3
6
-1
-1
3

总结：

我总是很难一次性想到通解，往往是根据样例再一点点修改，感觉最后得到的代码也很冗杂，思路不清晰，这个还没有完全通过样例。我是小笨蛋。有没有大佬能指点一下啊。

#include<iostream>
using namespace std;
//从上往下建树//n1和n2在改变树的高度 
int check(int a,int b,int c){
	int h=0;
	if(a>0)
	{
		for(h=0;h<a;h++){
			int pn=1;
			int i=h;
			while(i){
				pn*=2;
				i--; 
			}
			if(a<pn)break;
			a=a-pn;
		} 
		int t=1,p=0;//不改变层数的情况下，最多可以存放t个n0,p个n1 
		if(a==0){
			for(int i=0;i<h;i++){
				t*=2;
			}
			if(c!=t)h=-1;
			else if(b>0)h++;
		}
		if(a!=0)
		{
			for(int i=0;i<h;i++){	
				t*=2;
		}
			p=t-a; 
			t=t+a;
			if(c!=t)h=-1;
			else{
				h++;
				if(b>p){
					h++;
					b=b-p;
				}
				b=b-t;
				while(b>0){
					h++;
					b=b-t;
				}
			}
		}	
	} 
	else if(a==0){
		if(b==0)h=0;
	    else if(b!=0)h=b;
	    else h=-1;
	}
	return h;
}
int main(){
	int n;
	cin>>n;
	while(n--){
		int a,b,c;
		cin>>a>>b>>c;
		int ans=check(a,b,c);
		cout<<ans<<endl;
	}
}