携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第29天,点击查看活动详情
【LeetCode每日一题】998.最大二叉树 II
题目难度:简单
题意
最大树 定义:一棵树,并满足:其中每个节点的值都大于其子树中的任何其他值。
给你最大树的根节点 root 和一个整数 val 。
就像 之前的问题 那样,给定的树是利用 Construct(a) 例程从列表 a(root = 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) 。
示例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, 100]
内 1 <= Node.val <= 100
- 树中的所有值 互不相同
1 <= val <= 100
题解:递归+思维
如果val值大于根节点,则直接将原来的树的整棵树作为val节点的左子树即可。
反之val值节点一定在当前树的右子树中。 假设当前我们遍历的节点为cur,其父节点为par。当cur节点的值小于val时,构造新节点以val为值,且以cur为左子树。将该节点作为par新的右节点,并返回跟节点即可
当然还有最差的情况:如果遍历完成之后,仍然没有找到比val值小的节点,那么同样构造一个新的节点,以val为值,将该节点作为par的右节点,同样返回根节点即可。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* insertIntoMaxTree(TreeNode* root, int val) {
TreeNode* par=nullptr;
TreeNode* cur=root;
while(cur){
if(val > cur->val){
if(!par){
//表示val值大于root节点的值
return new TreeNode(val,cur,nullptr);
}
//在root的右子树中
TreeNode* node = new TreeNode(val,cur,nullptr);
par->right = node;
return root;
}else{
par=cur;
cur=cur->right;
}
}
//最差的情况:val节点在整棵二叉树的最后边节点
par->right=new TreeNode(val);
return root;
}
};