leetcode python3算法-根据一棵树的前序遍历与中序遍历构造二叉树。

最新推荐文章于 2022-06-19 16:39:04 发布

acarsar

最新推荐文章于 2022-06-19 16:39:04 发布

阅读量982

点赞数

CC 4.0 BY-SA版权

分类专栏：算法-Python实现文章标签： python 二叉树算法 LeetCode

本文链接：https://blog.youkuaiyun.com/acarsar/article/details/87927919

算法-Python实现专栏收录该内容

11 篇文章

订阅专栏

该博客详细介绍了如何根据一棵树的前序遍历和中序遍历结果，利用Python3来构造对应的二叉树。通过解析前序遍历和中序遍历的特性，确定根节点及其左右子树，并采用递归方式实现二叉树的构建。

1.题目描述
注意:
你可以假设树中没有重复的元素。

例如，给出

前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树：
在这里插入图片描述
2.思路

二叉树的前序遍历：规则是若二叉树为空，则空操作返回，否则先访问根结点，然后前序遍历左子树，再前序遍历右子树

遍历的顺序为：ABDGH-CEIF
二叉树的中序遍历：规则是若树为空，则空操作返回，否则从根结点开始（注意并不是先访问根结点），中序遍历根结点的左子树，然后是访问根结点，最后中序遍历右子树

遍历的顺序为：GDHBAE-ICF
所以前序的第一个节点必定是根节点，在中序遍历中找到根节点所在的索引x，观察研究可以发现，在中序遍历中根节点索引之前的即为其左孩子，即左孩子们 = root[:x];那么在前序遍历中的左孩子个数和中序遍历是一样的，即为从前序遍历中索引为1（索引0为根节点）到x+1的节点；其余的则为右孩子们。
最后递归调用自己，得到二叉树
3.代码

# 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
        """
        if inorder==[]:
            return None
        root = TreeNode(preorder[0])
        # 找到根节点在中序遍历中的位置(索引)
        index = inorder.index(root.val)
        # 递归调用
        root.left=self.buildTree(preorder[1:index+1],inorder[0:index])
        root.right=self.buildTree(preorder[index+1:],inorder[index+1:])
        return root