如何从前序与中序遍历序列构造python二叉树 - 大数据

博主:xiaoweixiaowei 2023-01-18 条评论

如何从前序与中序遍历序列构造python二叉树,很多新手对此不是很清楚,为了帮助大家解决这个难题,下面小编将为大家详细讲解,有这方面需求的人可以来学习下,希望你能有所收获。

【题目】

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

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

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

 

【思路】

首先回顾遍历的顺序:前序遍历是根节点-左子树-右子树,中序遍历是左子树-根节点-右子树。

那么前序遍历数组的第一个元素肯定是根节点,在中序遍历数组中找到这个元素,则其前一部分是左子树的元素,其后一部分是右子树的元素。递归即可求解。

注意:前序遍历+后序遍历,不能确定唯一的二叉树!

如何从前序与中序遍历序列构造python二叉树 - 大数据

【代码】

python版本

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

class Solution(object):
    def buildTree(self, preorder, inorder):
        """
        :type preorder: List[int]
        :type inorder: List[int]
        :rtype: TreeNode
        """
        # 前序遍历,第一个是head
        # 中序遍历,前一部分是左子树,后一部分是右子树
        if len(preorder) == 0:
            return None
        
        node = TreeNode(preorder[0])
        index = inorder.index(preorder[0])
        node.left = self.buildTree(preorder[1: index + 1], inorder[:index])
        node.right = self.buildTree(preorder[index + 1:], inorder[index + 1:])
        return node

【相似题目】

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

解题思路:后序遍历数组的最后一个元素是根节点的元素,同样在中序遍历数组中找到该元素,递归生成二叉树。

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

解题思路:直接生成只有右孩子的二叉树即可满足条件。

看完上述内容是否对您有帮助呢?如果还想对相关知识有进一步的了解或阅读更多相关文章,请关注云行业资讯频道,感谢您对云的支持。

The End

发布于:2023-01-18,除非注明,否则均为 主机评测原创文章,转载请注明出处。