代码随想录二叉树——二叉搜索树的插入操作

题目

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

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

返回：

或者

思路

其实可以不考虑题目中提示所说的改变树的结构的插入方式，二叉搜索树的插入不用考虑重构问题。

只需要按照二叉树搜索树的规则去遍历，然后遇到空节点就直接插入就可以了。

● 递归法（三部曲）：

1. 递归函数参数和返回值：参数为根节点和待插入元素值，返回值为节点类型
2. 终止条件：找到空节点即遍历到null的位置，就是要插入节点的位置了，并把插入的节点返回（返回是为了方便父子关系赋值，如将返回的值用root.left接住即可完成父子节点关系赋值）
3. 确定单层递归的逻辑：不需要遍历整棵树，因为这是搜索树，遍历整棵搜索树是对搜索树的侮辱（doge），搜索树是有方向的，可以根据插入元素的数值，决定递归方向。

class Solution {
	public TreeNode insertIntoBST(TreeNode root , int val){
		if(root == null)
			return new TreeNode(val);//如果当前节点为空，也就意味着val找到了合适的位置，此时创建节点直接返回。
		
		if(root.val < val){//这里通过接住递归函数返回值，来完成父子节点的赋值操作，如root.right表示让root的左子树接住递归的返回值insertIntoBST(root.right,val)，从而完成插入操作即让父节点指向孩子的操作
			root.right = insertIntoBST(root.right,val);//递归创建右子树 
		}
		if(root.val > val){
			root.left = insertIntoBST(root.left,val);//递归创建左子树
		}
		return root;
	}
}

● 迭代法：
在迭代法遍历的过程中，需要记录一下当前遍历节点的父节点，这样才能做插入节点的操作（即让root指向孩子）

class Solution {
	public TreeNode insertIntoBST(TreeNode root, int val){
		if(root == null) return new TreeNode(val);//如果当前节点为空，也就意味着val找到了合适的位置，此时创建节点直接返回。
		TreeNode newRoot = root;//方便操作
		TreeNode pre =root;
		while(root != null){
			pre = root;//从根节点开始搜索BST，根据有序性找到插入的位置的前驱位置，并且让pre指向这个位置,放在if语句前面是为了可以指向前驱位置
			if(root.val > val){
				root = root.left;
			}else if (root.val < val){
				root = root.right;
			}
		}
		//找到前驱节点之后，继续判断是插在左边还是右边
		if(pre.val > val){
			pre.left = new TreeNode(val);
		} else {
			pre.right = new TreeNode(val);
		}
		return newRoot;
	}
}