998. Maximum Binary Tree II

最新推荐文章于 2022-01-31 01:28:45 发布

原创最新推荐文章于 2022-01-31 01:28:45 发布 · 366 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

二叉树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了解决在已有的最大树中插入一个新值并构建新最大树的问题。提供了两种解决方案，一种是通过中序遍历后重新构建，另一种是在原树上直接寻找插入位置。代码实现使用C++。

We are given the root node of a maximum tree: a tree where every node has a value greater than any other value in its subtree.

Just as in the previous problem, the given tree was constructed from an list A (root = Construct(A)) recursively with the following Construct(A) routine:

If A is empty, return null.
Otherwise, let A[i] be the largest element of A. Create a root node with valueA[i].
The left child of root will be Construct([A[0], A[1], ..., A[i-1]])
The right child of root will be Construct([A[i+1], A[i+2], ..., A[A.length - 1]])
Return root.

Note that we were not given A directly, only a root node root = Construct(A).

Suppose B is a copy of A with the value val appended to it. It is guaranteed that Bhas unique values.

Return Construct(B).

Example 1:

Input: root = [4,1,3,null,null,2], val = 5
Output: [5,4,null,1,3,null,null,2]
Explanation: A = [1,4,2,3], B = [1,4,2,3,5]

Example 2:

Input: root = [5,2,4,null,1], val = 3
Output: [5,2,4,null,1,null,3]
Explanation: A = [2,1,5,4], B = [2,1,5,4,3]

Example 3:

Input: root = [5,2,3,null,1], val = 4
Output: [5,2,4,null,1,3]
Explanation: A = [2,1,5,3], B = [2,1,5,3,4]

题解：

在上一篇博客的基础上解的题

戳这里QAQ~~查看具体思路

大概就是在给定的一颗二叉树上，额外附加一个值val，建立另一颗二叉树，这个二叉树的建立规则请查看上一篇博客。

思路：

（一）、初始化一个vector用来存放给定的二叉树中序遍历出来的数值，然后在vector尾端加上val。然后再开始建立一颗新的二叉树。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoMaxTree(TreeNode* root, int val) {
        if(!root)
            return NULL;
        vector<int>res;
        Inorder(root,res);
        res.push_back(val);
        vector<TreeNode*>Root;
        for(int i=0;i<res.size();i++)
        {
            TreeNode* add=new TreeNode(res[i]);
            if(Root.empty())
                Root.push_back(add);
            else
            {
                while(!Root.empty()&&Root.back()->val<res[i])
                {
                    add->left=Root.back();
                    Root.pop_back();
                }
                if(!Root.empty())
                    Root.back()->right=add;
                Root.push_back(add);
            }
        }
        return Root.front();
    }
     void Inorder(TreeNode* root,vector<int>& res)
    {
        if(!root)
            return ;
        Inorder(root->left,res);
        res.push_back(root->val);
        Inorder(root->right,res);
    }
};

运行结果：

（二）、直接在原始的二叉树上寻找到val的插入点。

根据例子所示：附加值val是直接放在二叉树中序序列的最后的。所以：

假设：node代表当前节点，prev代表node的父节点

1.重复以下代码：

if（！node&&val<node->val）继续，else 结束。

2.，则val应该是prev的右孩子，node应该是val的左孩子。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoMaxTree(TreeNode* root, int val) {
       TreeNode *node=root,*prev=NULL;
        while(node&&val<node->val)
        {
            prev=node;
            node=node->right;
        }
        auto n=new TreeNode(val);
        if(prev)
          prev->right=n;
        else
            root=n;
        n->left=node;
    return root;
    }
};