题目
题目链接 给定一个不含重复元素的整数数组 nums 。一个以此数组直接递归构建的 最大二叉树 定义如下:
- 二叉树的根是数组 nums 中的最大元素。
- 左子树是通过数组中 最大值左边部分 递归构造出的最大二叉树。
- 右子树是通过数组中 最大值右边部分 递归构造出的最大二叉树。
返回有给定数组 nums 构建的 最大二叉树 。 示例 1:
输入:nums = [3,2,1,6,0,5]
输出:[6,3,5,null,2,0,null,null,1]
解释:递归调用如下所示:
- [3,2,1,6,0,5] 中的最大值是 6 ,左边部分是 [3,2,1] ,右边部分是 [0,5] 。
- [3,2,1] 中的最大值是 3 ,左边部分是 [] ,右边部分是 [2,1] 。
- 空数组,无子节点。
- [2,1] 中的最大值是 2 ,左边部分是 [] ,右边部分是 [1] 。
- 空数组,无子节点。
- 只有一个元素,所以子节点是一个值为 1 的节点。
- [0,5] 中的最大值是 5 ,左边部分是 [0] ,右边部分是 [] 。
- 只有一个元素,所以子节点是一个值为 0 的节点。
- 空数组,无子节点。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例 2:
输入:nums = [3,2,1]
输出:[3,null,2,null,1]
解题
方法一:递归
和leetcode-106:从中序与后序遍历序列构造二叉树的思路一模一样. 通过数组构造二叉树,都可以尝试这种方式
class Solution:
def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
def build(in_left,in_right):
if in_left>in_right:
return None
val = max(nums[in_left:in_right+1])
index = idx_map[val]
root = TreeNode(val)
root.right = build(index+1,in_right)
root.left = build(in_left,index-1)
return root
idx_map = {val:idx for idx,val in enumerate(nums)}
return build(0,len(nums)-1)
|