题目描述:
给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[3,2,1,6,0,5]
输出:返回下面这棵树的根节点:
6
/ \
3 5
\ /
2 0
\
1
提示:
给定的数组的大小在 [1, 1000] 之间。
个人思路:
根据最大二叉树的定义可知,本题可以使用递归解决。递归主体仅处理当前root节点,左右子树的构建交给递归去解决。
具体代码如下:
class Solution:
def constructMaximumBinaryTree(self, nums: List[int]) -> TreeNode:
if not nums:
return None
def helper(nums):
if not nums:
return None
max_index = nums.index(max(nums))
root = TreeNode(nums[max_index]) # 递归主体只对root节点进行最大树构造
root.left = helper(nums[:max_index])
root.right = helper(nums[max_index + 1:])
return root
return helper(nums)