华为OD机试(C卷,200分)- 二叉树计算

题目描述

给出一个二叉树如下图所示：

请由该二叉树生成一个新的二叉树，它满足其树中的每个节点将包含原始树中的左子树和右子树的和。

左子树表示该节点左侧叶子节点为根节点的一颗新树；右子树表示该节点右侧叶子节点为根节点的一颗新树。

输入描述

2行整数，第1行表示二叉树的中序遍历，第2行表示二叉树的前序遍历，以空格分割
例如：
7 -2 6 6 9
6 7 -2 9 6

输出描述

1行整数，表示求和树的中序遍历，以空格分割
例如：
-2 0 20 0 6

用例

输入 -3 12 6 8 9 -10 -7
8 12 -3 6 -10 9 -7
输出 0 3 0 7 0 2 0
说明 无

题目解析

1、前序数组为0，空节点
2、前序数组第一个元素为根节点
3、寻找根节点在中序数组位置作切割点
4、切中序数组
5、切前序数组
6、递归处理左区间右区间并计算每个节点的childSum，新树的每个节点的值 = 其左右子树的所有节点值之和。

#include <stdio.h>
#include <stdlib.h>
#define MAX 10000
int midOrder[MAX];
int preOrder[MAX];
typedef struct TreeNode{int val;//当前节点值int childSum;//当前节点左子树+右子树的和struct TreeNode *left;struct TreeNode *right;
}TreeNode;
TreeNode *new_TreeNode(int val){TreeNode *node=(TreeNode*)malloc(sizeof(TreeNode));node->val=val;node->childSum=0;node->left=NULL;node->right=NULL;return node;
}
void getMidOrder(TreeNode* root){if(root==NULL)return;getMidOrder(root->left);printf("%d ",root->childSum);getMidOrder(root->right);
}
struct TreeNode* buildTree(int* preorder, int preorderSize, int* inorder, int inorderSize) {if(!inorderSize)return NULL;struct TreeNode* node = (struct TreeNode*)malloc(sizeof(struct TreeNode));if (node == NULL) {// 处理内存分配失败return NULL;}node->val=preorder[0];int index;for (index = 0; index < inorderSize; index++) {if(inorder[index] == preorder[0]) {break;}}int rightSize=inorderSize-index-1;node->left=buildTree(preorder+1,index,inorder,index);//左闭右开node->right=buildTree(preorder+index+1,rightSize,inorder+index+1,rightSize);node->childSum = (node->left == NULL ? 0 : (node->left->val + node->left->childSum)) +(node->right == NULL ? 0 : (node->right->val + node->right->childSum));return node;
}
int main()
{int size=0;while(scanf("%d",&midOrder[size++])){if(getchar()!=' ')break;}for(int i=0;i<size;i++)scanf("%d",&preOrder[i]);TreeNode* root=buildTree(preOrder,size,midOrder,size);getMidOrder(root);return 0;
}