华为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,新树的每个节点的值 = 其左右子树的所有节点值之和。

c 复制代码
#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;
}
相关推荐
mit6.8242 分钟前
[Cyclone] 哈希算法 | SIMD优化哈希计算 | 大数运算 (Int类)
算法·哈希算法
c++bug5 分钟前
动态规划VS记忆化搜索(2)
算法·动态规划
哪 吒7 分钟前
2025B卷 - 华为OD机试七日集训第5期 - 按算法分类,由易到难,循序渐进,玩转OD(Python/JS/C/C++)
python·算法·华为od·华为od机试·2025b卷
军训猫猫头1 小时前
1.如何对多个控件进行高效的绑定 C#例子 WPF例子
开发语言·算法·c#·.net
success1 小时前
【爆刷力扣-数组】二分查找 及 衍生题型
算法
Orlando cron2 小时前
数据结构入门:链表
数据结构·算法·链表
牛客企业服务2 小时前
2025年AI面试推荐榜单,数字化招聘转型优选
人工智能·python·算法·面试·职场和发展·金融·求职招聘
呜喵王阿尔萨斯3 小时前
编程中的英语
c语言·c++
糖葫芦君3 小时前
Policy Gradient【强化学习的数学原理】
算法