998. 最大二叉树 II
难度中等
最大树定义:一个树,其中每个节点的值都大于其子树中的任何其他值。
给出最大树的根节点 root。
就像之前的问题那样,给定的树是从表 A(root = Construct(A))递归地使用下述 Construct(A) 例程构造的:
(1)如果 A 为空,返回 null
(2)否则,令 A[i] 作为 A 的最大元素。创建一个值为 A[i] 的根节点 root
(3)root 的左子树将被构建为 Construct([A[0], A[1], …, A[i-1]])
(4)root 的右子树将被构建为 Construct([A[i+1], A[i+2], …, A[A.length - 1]])
(5)返回 root
请注意,我们没有直接给定 A,只有一个根节点 root = Construct(A).
假设 B 是 A 的副本,并附加值 val。保证 B 中的值是不同的。
返回 Construct(B)。
示例 1:
输入: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:
输入: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:
输入: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.
1 <= B.length <= 100
思路1.0:
因为新出现的元素永远在数列的最右边,所以将其加入二叉树中的情况如下:
1)作为子节点时,只可能是右子树
2)作为父节点时,只拥有左子树
所以只需要遍历A树的最靠右的所有节点即可
代码1.0(已完成):
class Solution {
public:
TreeNode* insertValue(TreeNode* root, const int& val)
{
if (root != NULL)
{
if (val < root->val)
{
root->right = insertValue(root->right, val);
}
else
{
return new TreeNode(val, root, NULL);
}
}
else
{
return new TreeNode(val);
}
return root;
}
TreeNode* insertIntoMaxTree(TreeNode* root, int val) {
root = insertValue(root, val);
return root;
}
};