Leetcode 998.最大二叉树 II
1 题目描述(Leetcode题目链接)
最大树定义:一个树,其中每个节点的值都大于其子树中的任何其他值。给出最大树的根节点 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).
假设 B 是 A 的副本,并附加值 val。保证 B 中的值是不同的。
返回 Construct(B)。
输入: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 题解
最大二叉树题目链接,本题就是在 A A A数组后面加上一个 v a l val val后,树会变成什么样子。所以就是简单的查找和插入的问题,根据最大二叉树的构造原则,可以看出 v a l val val是应该处于右子树的,所以在右孩子中找第一个小于 v a l val 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:
new = TreeNode(val)
if root.val < val:
new.left = root
return new
node = root
while node.right and node.right.val > val:
node = node.right
new.left = node.right
node.right = new
return root