给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
- 二叉树的根是数组中的最大元素。
- 左子树是通过数组中最大值左边部分构造出的最大二叉树。
- 右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
Example 1:
输入: [3,2,1,6,0,5] 输入: 返回下面这棵树的根节点: 6 / \ 3 5 \ / 2 0 \ 1
注意:
- 给定的数组的大小在 [1, 1000] 之间
思路:
递归就完事了,根据题目要求不断递归形成子树。
class Solution(object):
def constructMaximumBinaryTree(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""
if not nums:
return None
root = TreeNode(max(nums))
root.left = self.constructMaximumBinaryTree(nums[:nums.index(root.val)])
root.right = self.constructMaximumBinaryTree(nums[nums.index(root.val)+1:])
return root