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

递归法:

首先考虑树为空的情况,只要返回一个值为val的结点即可;

当树不为空时,插入的算法为:

  1. 当val小于root.val,往下寻找左子树,即用root.left递归;
  2. 当val大于root.val(val值一定不等),往下寻找右子树,即用root.right递归;
  3. 直到找到一个null,说明可以在此插入值为val的结点了。

有两个地方需要去考虑,一是插入前记录父节点的位置,二是记录上一个递归是左儿子还是右儿子。父节点位置可以用一个全局变量记录一下;至于是左儿子还是右儿子可以用一个flag表示。在确定下一层是null时,把当前节点记录,并且把应插在左还是右也记录。

实现代码如下:

java 复制代码
class Solution {
 public TreeNode insertIntoBST(TreeNode root, int val) {
    	if(root==null) {
    		TreeNode ins = new TreeNode(val);
    		return ins;
    	}
    	insertBST(root,val);
    	return root;
    }
    
    int flag=0;//全局变量,0代表左儿子,1代表右儿子
    TreeNode father = new TreeNode();//全局变量记录父节点位置
    public void insertBST(TreeNode root,int val) {
    	if(root==null) {
    		TreeNode ins = new TreeNode(val);
    		if(flag==0) {
    			father.left=ins;
    		}else {
    			father.right=ins;
    		}
    		return;
    	}
    	if(val<root.val) {
    		if(root.left==null) {
    			father=root;
    		}
    		insertBST(root.left,val);
    	}
    	else {
    		if(root.right==null) {
    			father=root;
    			flag=1;
    		}
    		insertBST(root.right,val);
    	}
    }
}
相关推荐
AI柠檬31 分钟前
机器学习:数据集的划分
人工智能·算法·机器学习
让我们一起加油好吗1 小时前
【数论】裴蜀定理与扩展欧几里得算法 (exgcd)
算法·数论·裴蜀定理·扩展欧几里得算法·逆元
Geo_V1 小时前
提示词工程
人工智能·python·算法·ai
侯小啾1 小时前
【22】C语言 - 二维数组详解
c语言·数据结构·算法
TL滕2 小时前
从0开始学算法——第一天(如何高效学习算法)
数据结构·笔记·学习·算法
傻童:CPU2 小时前
DFS迷宫问题
算法·深度优先
B站_计算机毕业设计之家2 小时前
计算机视觉:python车辆行人检测与跟踪系统 YOLO模型 SORT算法 PyQt5界面 目标检测+目标跟踪 深度学习 计算机✅
人工智能·python·深度学习·算法·yolo·目标检测·机器学习
sheepfagdng2 小时前
求职专栏-【面试-自我介绍】
面试·职场和发展
一个不知名程序员www2 小时前
算法学习入门---前缀和(C++)
c++·算法
jackzhuoa2 小时前
Rust API 设计的零成本抽象原则:从语言基石到工程实践
算法·rust