LeetCode 654. 最大二叉树

题目

给定一个不重复的整数数组 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 的节点。
    • 空数组，无子节点。
      

方法

思路同 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