飞道的博客

从二叉树的前、中、后序遍历中构造二叉树及python实现。leetcode105、leetcode106、leetcode889(python题解)

404人阅读  评论(0)

问题:从二叉树的前、中、后遍历中构造二叉树及其python实现,整理了当模板。对遍历顺序疑惑的推荐另一个博主的文章:
https://blog.csdn.net/qq_33243189/article/details/80222629

一、从前序与中序遍历序列构造二叉树(leetcode105)
问题

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/

难度:中等

分析
前序遍历的第一个节点是根节点,从中序遍历中找到根节点的位置,得到左树和右树的大小,在前序遍历序列中找到左树和右树。如此循环。

解决方法
1:递归

#递归
#超99%
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        def myBuildTree(preorder_left, preorder_right, inorder_left, inorder_right):
            if preorder_left > preorder_right: return 
            preorder_root = preorder_left
            inorder_root = index[preorder[preorder_root]]
            root = TreeNode(preorder[preorder_root])
            size_left_subtree = inorder_root - inorder_left
            root.left = myBuildTree(preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1)
            root.right = myBuildTree(preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right)
            return root
        n = len(preorder)
        index = {
   element: i for i, element in enumerate(inorder)}
        return myBuildTree(0, n - 1, 0 ,n - 1)

二、从中序与后序遍历序列构造二叉树(leetcode106)
问题

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-inorder-and-postorder-traversal/

难度:中等

分析
思路与从前序和中序遍历序列构造二叉树相同,只是后序序列中的根节点值是从后面开始取的。

解决方法
1:递归

class Solution:
    def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
        def helper(in_left,in_right):
            if in_left > in_right:
                return 
            val = postorder.pop()
            root = TreeNode(val)
            index = idx_map[val]
            root.right = helper(index + 1, in_right)
            root.left = helper(in_left, index - 1)
            return root
        idx_map = {
   val: idx for idx, val in enumerate(inorder)}
        return helper(0,len(inorder) - 1)

三、根据前序和后序遍历构造二叉树(leetcode889)
问题

题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-postorder-traversal/

难度:中等

分析
前序遍历的根节点后的第一个值是左子树的根节点,后序遍历序列中左子树最后一个值是根节点,在后序遍历中找到左子树根节点的位置,就确定的左子树的大小,因此得以确定右子树的大小。

解决方法
1:递归

class Solution(object):
    def constructFromPrePost(self, pre, post):
        if not pre: return None
        root = TreeNode(pre[0])
        if len(pre) == 1: return root

        L = post.index(pre[1]) + 1
        root.left = self.constructFromPrePost(pre[1:L+1], post[:L])
        root.right = self.constructFromPrePost(pre[L+1:], post[L:-1])
        return root

转载:https://blog.csdn.net/AI414010/article/details/108919653
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场