leetcode—20.二叉树构建相关题目leetcode总结

引言

  这类题目本质还是用递归的手法来实现，但是这类题目有一个特点，就是它是构建一棵树，而不是给定一棵树，然后进行遍历，所以实现起来思路上有点逆向。实现思路是在递归中创建根节点，然后找到将元素劈成左右子树的方法，递归得到左右根节点，接上创建的根然后返回。

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

根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

示例：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if not(preorder and inorder):
            return None
        # 从前序数组中找根节点
        root = TreeNode(preorder[0])
        # 寻找根节点在中序数组中的索引
        mid_index = inorder.index(preorder[0])
        # 将树分为左子树与右子树,递归处理左子树与右子树
        root.left = self.buildTree(preorder[1:mid_index+1],inorder[:mid_index])
        root.right = self.buildTree(preorder[mid_index+1:],inorder[mid_index+1:])
        return root

106. 从中序与后序遍历序列构造二叉树

根据一棵树的中序遍历与后序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

示例：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def buildTree(self, inorder: List[int], postorder: List[int]) -> TreeNode:
        # 如果中序与后序为空
        if not (inorder and postorder):
            return None

        # 从后序遍历数组中找根节点
        root = TreeNode(postorder[-1])
        # 找到在中序数组中根节点的索引
        mid_index = inorder.index(postorder[-1])
        # 将树分为左子树与右子树，递归调用左子树与右子树
        root.left = self.buildTree(inorder[:mid_index],postorder[:mid_index])
        root.right = self.buildTree(inorder[mid_index+1:],postorder[mid_index:-1])
        return root

889. 根据前序和后序遍历构造二叉树

  我们知道前序+中序，中序+后序能够唯一确定一颗二叉树。前序+后序可以确定多颗二叉树。这里给出了前序+后序确定二叉树的一种解。

返回与给定的前序和后序遍历匹配的任何二叉树。
pre 和 post 遍历中的值是不同的正整数。

思路：

确定根节点后，再确定左子树与右子树的范围

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

class Solution:
    def constructFromPrePost(self, pre: List[int], post: List[int]) -> TreeNode:
        def dfs(pre,post):
            if not pre:
                return None
            # 如果pre数组长度为1，直接返回即可
            if len(pre) == 1:
                return TreeNode(pre[0])

            # 前序遍历数组的第一个为根节点
            root = TreeNode(pre[0])
            # 在后序数组中寻找前序数组的第二个节点（左子树的根节点）,+1即为左子树的数量
            left_count = post.index(pre[1])+1
            # 递归遍历左子树与右子树
            root.left = dfs(pre[1:left_count+1],post[:left_count])
            root.right = dfs(pre[left_count+1:],post[left_count:-1])
            return root
        return dfs(pre,post)

---

如果对您有帮助，麻烦点赞关注，这真的对我很重要！！！如果需要互关，请评论或者私信！

---