题意:求二叉树垂直方向节点的和。
思路:由于题目给出的数据是前序遍历的,最直接的方法是建树模拟。最简单的方法是用数组模拟求和。我用的是数组模拟,以根节点为中心,遇到左树-1,遇到右树+1,向两边扩展,利用建树的递归思想。
#include <iostream>
#include <cstring>
using namespace std;
int ans[100];
void build(int, int);
int main()
{
//freopen("data.txt", "r", stdin);
int x;
int cases = 1;
while (scanf("%d", &x) && x != -1)
{
printf("Case %d:\n", cases++);
memset(ans, 0, sizeof(ans));
build(50, x);//从中间往两边扩展
int flag = 0;
for(int i = 0; i < 100; i++)
{
if(ans[i] != 0 && flag == 0)
{
printf("%d", ans[i]);
flag = 1;
}
else if(ans[i] != 0 && flag == 1)
printf(" %d", ans[i]);
}
printf("\n\n");
}
return 0;
}
void build(int n, int num)
{
ans[n] += num;
int s;
scanf("%d", &s);
if(s != -1)
build(n - 1, s);//读取左孩子
scanf("%d", &s);
if(s != -1)
build(n + 1, s);//读取右孩子
return;
}