题目
给定一个不重复的整数数组 nums 。最大二叉树可以用下面的算法从 nums 递归地构建:
- 创建一个根节点，其值为 nums 中的最大值。
- 递归地在最大值左边的子数组前缀上构建左子树。
- 递归地在最大值右边的子数组后缀上构建右子树。
返回 nums 构建的最大二叉树。
例：
输入：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 的节点。
-
空数组，无子节点。
-
[3,2,1] 中的最大值是 3 ，左边部分是 [] ，右边部分是 [2,1] 。
方法
思路同 105. 从前序与中序遍历序列构造二叉树 和 106. 从中序与后序遍历序列构造二叉树
- 判断数组是否为空，即二叉树是否为空
- 取数组中的最大值作为根节点的值，构造根节点
- 找到根节点的值在数组 nums 中的索引值，通过该值将数组 nums 分成左右子树两个部分
- 不断调用 constructMaximumBinaryTree 函数，构造二叉树
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution(object):
def constructMaximumBinaryTree(self, nums):
if not nums:
return None
root_val = max(nums)
root = TreeNode(root_val)
separator = nums.index(root_val)
left = nums[: separator]
right = nums[separator + 1:]
root.left = self.constructMaximumBinaryTree(left)
root.right = self.constructMaximumBinaryTree(right)
return root
