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

最新推荐文章于 2024-10-01 22:51:02 发布

chenz1hao

最新推荐文章于 2024-10-01 22:51:02 发布

阅读量83

点赞数 1

文章标签：二叉树数据结构算法 leetcode 面试

本文链接：https://blog.csdn.net/qq_29941979/article/details/116541955

版权

这篇博客讨论了如何根据先序和中序遍历来构建二叉树的问题。作者指出，虽然递归方法的实现更直观，但在面试时可能会面临写出无bug代码的挑战。代码示例展示了如何通过递归函数构建二叉树节点，首先找到根节点，然后递归地处理左子树和右子树。

摘要由CSDN通过智能技术生成

问题描述：

思路简述：

以前在学数据结构的时候遇到过这种题目，但是说实话没有用代码来实现过，迭代和递归都可以构造出来，但是相比来说递归构造的代码更加直观容易理解一些，思路就是找到先序遍历中的第一个元素（也就是最终整个二叉树的根节点），通过该根节点，得到其在中序遍历中的下标，比如用k表示，然后分别得到左右子树的先序和中序遍历，左子树的先序遍历也就是preorder[1:1+k]，左子树的中序遍历也就是inorder[0:k]；右子树的先序遍历为：preorder[1+k:]，右子树的中序遍历为：inorder[1+k:]。

这样的题目看似基础简单，但是如果没有练习过，可能在面试当场的时候还比较难马上给出一段bugfree的代码。

代码实现：

class Solution:
    def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
        if len(preorder) == 0:
            return None
        res = TreeNode(preorder[0])
        
        k = inorder.index(preorder[0])
        
        preorderLeft = preorder[1:k+1]
        inorderLeft = inorder[0:k]
        res.left = self.buildTree(preorderLeft, inorderLeft)

        preorderRight = preorder[k+1:]
        inorderRight = inorder[k+1:]
        res.right = self.buildTree(preorderRight, inorderRight)

        return res