LeetCode701. 二叉搜索树中的插入操作-递归与迭代解法-100%+83%(超高效的解法!!!!)

https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/

题意

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

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

在这里插入图片描述

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

在这里插入图片描述

题解

往二叉搜索树中插入元素,且插入的元素与树中的元素不相同。
迭代解法使用常规思路,找到插入的位置,创建新节点与父节点相连,只不过是作为左节点还是右节点需要另外判断。

迭代解法

/**
* 迭代解法
 */
class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root==null)//树为空
            return new TreeNode(val);
        TreeNode node=root;
        while(true){
            if(val<node.val){
                if(node.left!=null){//若还有左子节点,则继续迭代
                    node=node.left;
                }else {//否则就应放在当前节点左侧
                    node.left = new TreeNode(val);
                    return root;
                }
            }else {
                if(node.right!=null) {//若还有右子节点,则继续迭代
                    node = node.right;
                }else {//否则就应放在当前节点右侧
                    node.right = new TreeNode(val);
                    return root;
                }
            }
        }
    }
}

在这里插入图片描述

递归解法

递归解法,先明确insertIntoBST方法的功能,该方法应该在根节点为root的树中插入元素为val的新节点,并返回插入完成后的头节点。假如对root的左节点使用该方法,则表示在根节点为root.left的树中插入元素为val的新节点,并返回插入完成后的头节点。此时,左子树的头节点可能因插入操作而产生变化,在此需要更新root的左节点(比如左子树本来为空,插入多了一个节点),右子树也是相同的操作。

root.left=insertIntoBST(root.left,val);
root.right=insertIntoBST(root.right,val);

所以只需要找到新节点插入的大致方向(左或者右),然后交给左子树或右子树去处理即可。
再来考虑边界条件:什么找到真正该插入的位置?将插入操作不断的交给左、右子树去处理,当左或右节点为空时,就是应该插入的位置。此时只需要创建一个新节点返回,交给父节点跟新即可。

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if(root==null)
            return new TreeNode(val);
        if(val<root.val)//在左子树中插入并跟新左节点
            root.left=insertIntoBST(root.left,val);
        else在右子树中插入并跟新左节点
            root.right=insertIntoBST(root.right,val);
        return root;
    }
}

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值