leetcode_105. 从前序与中序遍历序列构造二叉树

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

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

例如,给出

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

    3
   / \
  9  20
    /  \
   15   7

二、解题思路

前序遍历:根->左->右,中序遍历:左->根->右,因此从前序遍历每次都取第一个元素进行构造,同时也找到取出的值在中序遍历里的位置,那么根据中序遍历的规则,该值的位置左边就是左子树元素右边就是右子树元素,递归构造即可。

三、代码

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

class Solution:
    def buildTree(self, preorder: list, inorder: list) -> TreeNode:
        def dfs(preorder, inorder):
            if inorder:
                val = preorder.pop(0)
                i = inorder.index(val)
                node = TreeNode(val)
                node.left = dfs(preorder, inorder[:i])
                node.right = dfs(preorder, inorder[i + 1:])
                return node
        return dfs(preorder, inorder)


if __name__ == '__main__':
    preorder = [3,9,20,15,7]
    inorder = [9, 3, 15, 20, 7]

    s = Solution()
    ans = s.buildTree(preorder, inorder)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值