JZ4 重建二叉树 python

最新推荐文章于 2022-10-04 19:42:40 发布

Sher_lyj

最新推荐文章于 2022-10-04 19:42:40 发布

阅读量140

点赞数

分类专栏：剑指offer 编程题文章标签：二叉树 python 递归算法

本文链接：https://blog.csdn.net/qq_42187809/article/details/119607932

版权

编程题同时被 2 个专栏收录

56 篇文章 0 订阅

订阅专栏

剑指offer

48 篇文章 0 订阅

订阅专栏

JZ4 重建二叉树

在这里插入图片描述

在这里插入图片描述
考虑递归，注意前序遍历和后序遍历的序列在划分为子问题时，长度仍然保持相等
book method
错误记录

未考虑输入为空的特殊情况

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
class Solution:
    # 返回构造的TreeNode根节点
    def reConstructBinaryTree(self, pre, vin):
        if len(pre) != len(vin):
            return -1
        else:
            if pre: 
                Tree = TreeNode(pre[0])
                vin_idx = vin.index(pre[0])
                if vin_idx == 0:
                    Tree.left = None
                else:
                    Tree.left = self.reConstructBinaryTree(pre[1:1+vin_idx], vin[:vin_idx])
                if vin_idx == len(pre)-1:
                    Tree.right = None
                else:
                    Tree.right = self.reConstructBinaryTree(pre[vin_idx+1:], vin[vin_idx+1:len(vin)])
                return Tree
            return None