华为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;
}
相关推荐
一定要AK11 分钟前
2025—暑期训练一
算法
一定要AK15 分钟前
贪心专题练习
算法
森焱森1 小时前
无人机三轴稳定控制(2)____根据目标俯仰角,实现俯仰稳定化控制,计算出升降舵输出
c语言·单片机·算法·架构·无人机
ytttr8732 小时前
matlab通过Q学习算法解决房间路径规划问题
学习·算法·matlab
go54631584653 小时前
修改Spatial-MLLM项目,使其专注于无人机航拍视频的空间理解
人工智能·算法·机器学习·架构·音视频·无人机
油泼辣子多加3 小时前
【Torch】nn.BatchNorm1d算法详解
算法
nlog3n3 小时前
基于 govaluate 的监控系统中,如何设计灵活可扩展的自定义表达式函数体系
算法·go
IT古董3 小时前
【第三章:神经网络原理详解与Pytorch入门】01.神经网络算法理论详解与实践-(2)神经网络整体结构
pytorch·神经网络·算法
ThetaarSofVenice3 小时前
垃圾收集相关算法Test
java·jvm·算法
小陈phd4 小时前
langchain从入门到精通(二十八)——RAG优化策略(六)集成多种检索器算法实现混合检索及问题转换总结
算法