跳转至

LeetCode: 998. 最大二叉树 II

1、题目描述

最大树定义:一个树,其中每个节点的值都大于其子树中的任何其他值。

给出最大树的根节点 root

就像之前的问题那样,给定的树是从表 A(root = Construct(A))递归地使用下述 Construct(A) 例程构造的:

  • 如果 A 为空,返回 null
  • 否则,令 A[i] 作为 A 的最大元素。创建一个值为 A[i] 的根节点 root
  • root 的左子树将被构建为 Construct([A[0], A[1], ..., A[i-1]])
  • root 的右子树将被构建为 Construct([A[i+1], A[i+2], ..., A[A.length - 1]])
  • 返回 root 请注意,我们没有直接给定 A,只有一个根节点 root = Construct(A).

假设 BA 的副本,并附加值 val。保证 B 中的值是不同的。

返回 Construct(B)

示例 1:

img

输入:root = [4,1,3,null,null,2], val = 5
输出:[5,4,null,1,3,null,null,2]
解释:A = [1,4,2,3], B = [1,4,2,3,5]

示例 2:

img

输入:root = [5,2,4,null,1], val = 3
输出:[5,2,4,null,1,null,3]
解释:A = [2,1,5,4], B = [2,1,5,4,3]

示例 3:

img

输入:root = [5,2,3,null,1], val = 4
输出:[5,2,4,null,1,3]
解释:A = [2,1,5,3], B = [2,1,5,3,4]

提示:

  • 1 <= B.length <= 100

2、解题思路

  • 分三种情况讨论

  • 没有根节点

  • 根节点小于val
  • 需要找到插入点

找插入点的时候,可以直接从右节点寻找

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def insertIntoMaxTree(self, root: TreeNode, val: int) -> TreeNode:
        node = TreeNode(val)
        if not root:
            return node

        if root.val < val:
            node.left = root
            return node

        temp = root
        while temp.right and temp.right.val > val:
            temp = temp.right

        node.left = temp.right
        temp.right = node

        return root
  • 递归实现
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def insertIntoMaxTree(self, root: TreeNode, val: int) -> TreeNode:
        node = TreeNode(val)
        if not root:
            return node

        if root.val < val:
            node.left = root
            return node

        root.right = self.insertIntoMaxTree(root.right, val)
        return root