654. 最大二叉树

给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:

二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。

示例 :

输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:

      6
    /   \
   3     5
    \    / 
     2  0   
       \
        1

算法步骤:

  • 首先调用 construct(nums, 0, n)
  • 在(l:r-1)内找到最大值的下标max_i,将 nums[max_i] 作为根节点的值。
  • 调用construct(nums, l, max_i)创建根节点的左孩子。递归执行此操作,创建根节点的左子树。
  • 同理,调用 construct(nums, max_i + 1, r) 创建根节点的右子树。
  • 方法 construct(nums, 0, n) 返回根节点。
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
        def max_(nums, l, r):
            max_i = l;
            for i in range(l,r):
                if nums[max_i] < nums[i]:
                    max_i = i
            return max_i
        def construct(nums, l, r):
            if l == r:
                return None
            max_i = max_(nums, l, r)
            root = TreeNode(nums[max_i])
            root.left = construct(nums, l, max_i)
            root.right = construct(nums, max_i + 1, r)
            return root
        return construct(nums, 0, len(nums));
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值