输入的二叉树是先序遍历的形式,只是加上了各个叶子节点的左右孩子,都为-1.
在递归的过程中算出水平位置的权值。(水平位置的权值,可以通过+1(右边节点)或者-1(左边节点) 来对应)
代码:
#include <iostream>
#include <cstring>
using namespace std;
const int maxn = 10000 + 10;
int sum[maxn];
void build(int pos)
{
int v;
cin >> v;
if(v == -1)
return;
sum[pos] += v;
build(pos - 1);
build(pos + 1);
}
bool init()
{
int v;
cin >> v; //确定根节点
if(v == -1)
return false;
memset(sum, 0, sizeof(sum));
sum[maxn / 2] = v;
build(maxn / 2 - 1);//左子树
build(maxn / 2 + 1);//右子树
}
int main()
{
int cases = 0;
while(init())
{
int i = 0;
while(sum[i] == 0)
i++;
cout << "Case " << ++cases << ":" << endl;
cout << sum[i++];
while(sum[i] != 0)
cout << " " << sum[i++];
cout << "\n\n";
}
return 0;
}