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

描述

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

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

分析

遍历搜索二叉树,当遇到null时,说明找到了符合的位置,new一个新结点返回就好。 不要全遍历搜索二叉树,利用搜索二叉树的特性去遍历。也可以迭代实现,需要一个变量记录要插入结点的父节点。

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