[LeetCode] 105. Construct Binary Tree from Preorder and Inorder Traversal @ python

最新推荐文章于 2021-03-18 16:18:19 发布

原创最新推荐文章于 2021-03-18 16:18:19 发布 · 131 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode

收录于

leetcode 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一种从给定的前序遍历和中序遍历序列还原二叉树的方法。通过理解两种遍历方式的特点，使用递归算法构建二叉树。代码示例展示了如何利用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
        """
        if not preorder and not inorder:
            return None
        root = TreeNode(preorder[0])
        index = inorder.index(preorder[0])
        root.left = self.buildTree(preorder[1:index+1],inorder[0:index])
        root.right = self.buildTree(preorder[index+1:],inorder[index+1:])
        return root