Leetcode# 701. 二叉搜索树中的插入操作

701. 二叉搜索树中的插入操作

给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。

注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

示例 1:

输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是:

示例 2:

输入:root = [40,20,60,10,30,50,70], val = 25 输出:[40,20,60,10,30,50,70,null,null,25] 示例 3:

输入:root = [4,2,7,1,3,null,null,null,null,null,null], val = 5 输出:[4,2,7,1,3,5]

提示:

给定的树上的节点数介于 0 和 10^4 之间 每个节点都有一个唯一整数值,取值范围从 0 到 10^8 -10^8 <= val <= 10^8 新值和原始二叉搜索树中的任意节点值都不同

本题第一想法用递归解决,思路为判断结点是否为NULL,为空则在此处插入待插入结点。不空则与待插入val值比较,若大于当前节点中的值,则向当前节点right递归调用自身函数,小于同理。(赋第一次代码(有问题))

class Solution {
          
   
public:
    void insert(TreeNode* root, int val){
          
   
        if(!root){
          
   
        	root = new TreeNode(val);
            return;
        }
        else if(root -> val > val)  insert(root -> left, val);
        else if(root -> val < val)  insert(root -> right, val);
        
    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
          
           
        insert(root, val);    
        return root;
    }
};

开始时遇到问题,递归完成后新建结点没有插入树中,测试时输出中无插入的节点。 后发现问题在于新建的结点没有与树连接,从树中无法访问到插入的节点。 之后对其进行修改,删去void函数,直接在TreeNode* insertIntoBST函数中实现操作。因为这样有返回值,可以通过将新建的结点返回并赋值给他的父节点得左子树或右子树。(赋修改后代码(无问题))

class Solution {
          
   
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
          
           
        if(!root){
          
   
            return new TreeNode(val);
        }
        else if(root -> val > val)  root -> left = insertIntoBST(root -> left, val);
        else if(root -> val < val)  root -> right = insertIntoBST(root -> right, val);   
        return root;
    }
};
经验分享 程序员 微信小程序 职场和发展