华为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;
}
相关推荐
OEC小胖胖1 小时前
渲染篇(二):解密Diff算法:如何用“最少的操作”更新UI
前端·算法·ui·状态模式·web
找不到、了1 小时前
Java排序算法之<归并排序>
算法·排序算法
香蕉可乐荷包蛋1 小时前
排序算法 (Sorting Algorithms)-Python示例
python·算法·排序算法
Sylvia-girl2 小时前
排序查找算法,Map集合,集合的嵌套,Collections工具类
java·算法·排序算法
TT哇2 小时前
【分治】归并排序——排序数组(medium)
java·算法·排序算法
aiprtem2 小时前
LVGL + ESP-Brookesia 嵌入式模拟桌面应用开发
linux·c语言·物联网
skyang.2 小时前
LeetCode 85. 最大矩形
算法·leetcode·职场和发展
滋滋不吱吱3 小时前
枚举中间位置基础篇
考研·算法·leetcode
阳光不锈@3 小时前
算法:最长递增子序列解法记录
算法·最长递增子序列·超详细分析·java实现
遇见尚硅谷3 小时前
C语言:20250728学习(指针)
c语言·开发语言·数据结构·c++·笔记·学习·算法