华为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;
}