LeetCode第105题—从前序与中序遍历序列构造二叉树—Python实现

Posted StriveZs

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode第105题—从前序与中序遍历序列构造二叉树—Python实现相关的知识,希望对你有一定的参考价值。


title: LeetCode No.105

categories:

  • OJ
  • LeetCode

tags:

  • Programing
  • LeetCode
  • OJ

LeetCode第105题—从前序与中序遍历序列构造二叉树

自己代码的开源仓库:click here 欢迎Star和Fork 😃

题目描述

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例如,给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:

    3
   / \\
  9  20
    /  \\
   15   7

代码

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
class Solution(object):
    def irritationBuildTree(self,preorder,inorder):
        if len(preorder) == 0: # 则证明没有根节点了
            return None
        curRoot = preorder.pop(0) # 获得当前子树根节点
        node = TreeNode(curRoot) # 生成节点
        index = inorder.index(curRoot) # 获得当前根节点在中序遍历中的位置,左边划分为左子树,右边为右子树
        node.left = self.irritationBuildTree(preorder[:index],inorder[:index]) # 递归构建左子树
        node.right = self.irritationBuildTree(preorder[index:],inorder[index+1:]) # 递归构建右子树
        return node

    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode

        核心思想:
            根据前序遍历和中序遍历来确定一颗二叉树
            1.确定树的根节点,树根是当前树中所有元素在前序遍历中最先出现的元素。

            2.求解树的子树,找出根节点在中序遍历中的位置,根左边的所有元素就是左子树,
            根右边的所有元素就是右子树。若根节点左边或右边为空,则该方向子树为空;
            若根节点左边和右边都为空,则根节点已经为叶子节点。

            3.递归求解树,将左子树和右子树分别看成一棵二叉树,重复1、2、3步,直到所有的节点完成定位。
        """
        return self.irritationBuildTree(preorder,inorder)

if __name__ == '__main__':
    s = Solution()
    print(s.irritationBuildTree([3,9,20,15,7],[9,3,15,20,7]))

以上是关于LeetCode第105题—从前序与中序遍历序列构造二叉树—Python实现的主要内容,如果未能解决你的问题,请参考以下文章

精选力扣500题 第46题 LeetCode 105. 从前序与中序遍历序列构造二叉树c++/java详细题解

LeetCode(105):从前序与中序遍历序列构造二叉树

LeetCode105. 从前序与中序遍历序列构造二叉树

LeetCode105. 从前序与中序遍历序列构造二叉树

leetcode: 105. 从前序与中序遍历序列构造二叉树

Leetcode 105. 从前序与中序遍历序列构造二叉树