已知前中遍历,重建二叉树,Python

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:
            return None
        # 拿到根节点
        root_val = preorder[0]
        # 创建节点
        root = TreeNode(root_val)
        # 拿到根节点在中序遍历中的位置,以确定左右子树个数
        root_idx = inorder.index(root_val)
        # 分割左右子树,进行递归操作
        l = self.buildTree(preorder[1:root_idx+1], inorder[0:root_idx])
        r = self.buildTree(preorder[root_idx+1:], inorder[root_idx+1:])
        # 将得到的左右子树添加到根节点的左右节点
        root.left = l
        root.right = r
        
        # 返回根节点
        return root

  • 3
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值