LeetCode:889. Construct Binary Tree from Preorder and Postorder Traversal - Python

问题描述:

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

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

示例:

输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]

提示:

  • 1 <= pre.length == post.length <= 30
  • pre[]post[]都是 1, 2, ..., pre.length 的排列
  • 每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。

问题分析:

我们知道,中序和先序、后续,可以唯一确定一个二叉树,而先序和后续,就不一定了。因为返回一个即可,所以不用考虑太繁杂。从整个序列来看看,pre第一个元素和post最后一个元素是根节点,所以,可以从pre开始,进行遍历,依次作为根节点,与此同时,确定一个这个根节点在post中的位置,获取两个区间,左右深度递归。

Python3实现:

class Solution:
    def constructFromPrePost(self, pre, post):
        if not pre: return None
        root = TreeNode(pre[0])  # 创建当前根节点
        if len(pre) == 1: return root  # 如果只有一个值,直接返回

        i = post.index(pre[1]) + 1  # 获取当前跟节点,在 post 中的位置
        root.left = self.constructFromPrePost(pre[1:i+1], post[:i])  # 递归左子树
        root.right = self.constructFromPrePost(pre[i+1:], post[i:])  # 递归右子树
        return root

声明: 总结学习,有问题可以批评指正,大神可以略过哦。

猜你喜欢

转载自blog.csdn.net/XX_123_1_RJ/article/details/82081857