给定一个不含重复元素的整数数组。一个以此数组构建的最大二叉树定义如下:
二叉树的根是数组中的最大元素。
左子树是通过数组中最大值左边部分构造出的最大二叉树。
右子树是通过数组中最大值右边部分构造出的最大二叉树。
通过给定的数组构建最大二叉树,并且输出这个树的根节点。
示例 :
输入:[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));