题意:给一棵二叉树,每个结点都有一个水平位置:左子结点在它左边一个单位,右子结点在右边1个单位。从左向右输出每个水平位置的所有结点的权值之和。如图6-7所示,从左到右的3个位置的权和分别为7,11,3。按照递归(先序)方式输入,用-1表示空树。
思路:开一个数组维护该列的和即可,DFS的过程中边输入边统计。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 200;
int v, sum[MAXN];
void dfs(int pos)
{
scanf("%d", &v);
if (v != -1) sum [pos-1] += v, dfs(pos-1);
scanf("%d", &v);
if (v != -1) sum [pos+1] += v, dfs(pos+1);
}
int main()
{
int CASE = 1;
while (~scanf("%d", &v) && v != -1)
{
memset(sum, 0, sizeof(sum));
sum[MAXN/2] = v; dfs(MAXN/2);
printf("Case %d:\n", CASE++);
for (int i = 0; i < 200; i++)
{
if (sum[i]) printf("%d", sum[i]);
if (sum[i] && sum[i+1] != 0) printf(" ");
}
printf("\n\n");
}
return 0;
}
/*
5 7 -1 6 -1 -1 3 -1 -1
8 2 9 -1 -1 6 5 -1 -1 12 -1
-1 3 7 -1 -1 -1
-1
*/