【LeetCode每日一题】998.最大二叉树 II

79 阅读2分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第29天,点击查看活动详情

【LeetCode每日一题】998.最大二叉树 II

题目链接:leetcode.cn/problems/mi…

题目难度:简单

题意

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

给你最大树的根节点 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:

image.png

image.png

输入: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;
    }
};