华为OD机试 Python【生成求和二叉树】

题目

请由二叉树生成一个新的二叉树，它满足其树中的每个节点将包含原始树中的左子树和右子树的和。左子树表示该节点左侧叶子节点为根节点的一颗新树；右子树表示该节点右侧叶子节点为根节点的一颗新树。

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

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

解题思路

1、构建原二叉树：通过前序遍历和中序遍历数组，递归构建原二叉树。
2、生成求和二叉树：通过递归计算每个节点的左子树和右子树的和，并更新节点值。
3、中序遍历输出：对生成的求和二叉树进行中序遍历，收集节点值并输出。

代码

class TreeNode:
    <