UVA 699 The falling leaves 详细~

题目链接

题目大意:

给定一棵节点带权的二叉树,每个节点都有一个水平位置,左子节点在它左边一个单位,右子节点在它右边一个单位。要求从左往右依次输出每个水平位置上的节点权值和。

例如：

该二叉树水平方向上有3个位置，从左到右的权值和分别为7，11，3

具体的输入输出细节可参见原题

解题思路：

解题方法实际上在题目描述中已经给出，对于任意一个水平位置为$x$的节点，其左子节点的水平位置为$x-1$，右子节点水平位置为$x+1$，因此如果我们将根节点放在一个合适的位置，然后遍历整棵树，将各个位置的权值和求出即可。

那么这个合适的位置是多少呢？因为题目中已经说明输出的权值和个数不会超过80，因此我们可以开一个长度为 $2\ast 80+1$的数组，将根节点放在 $80（从0编号）$的位置，然后遍历整棵树，这样就能保证所有的节点的水平位置都不会越界。

关键代码如下：

#include <iostream>
#include <cstring> 
using namespace std;

const int maxn = 2*80+1;
int sum[maxn];	// 统计各个水平位置的权值和

void Count(int p) // 统计树根的水平位置为 p 的权值和
{
	int v; cin >> v;
	if(v==-1) return;
	sum[p] += v;
	Count(p-1);
	Count(p+1);
}

bool init()	// 边读边统计
{
	int v;
	cin >> v;
	if(v==-1) return false;
	memset(sum,0,sizeof(sum));
	int p = maxn >> 1;
	sum[p] = v;
	Count(p-1);
	Count(p+1);
	return true;
}

int main()
{
	int kase = 0;	// 测试组数
	while(init()) {
		int p = 0;
		while(!sum[p]) p++	;	// 寻找最左边的位置
		cout << "Case " << ++kase << ":\n" << sum[p++];
		while(sum[p]) cout << " " << sum[p++];
		cout << "\n\n";
	}
	return 0;
}