【力扣-数据结构】【第 13 天】701. 二叉搜索树中的插入操作

标题	701. 二叉搜索树中的插入操作
难度	中等
天数	第13天,第2/2题
数据结构	树

给定二叉搜索树（BST）的根节点 root 和要插入树中的值 value ，

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

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

示例 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, 104]的范围内。
• -108 <= Node.val <= 108
• 所有值 Node.val 是 独一无二 的。
• -108 <= val <= 108
• 保证 val 在原始BST中不存在。

以上内容来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

1. 注意这个是一个二叉搜索树（BST）

   • 左树 < 根 < 右树
   • 刚开始没注意，就随便找个叶子节点插入了，直接没通过

2. 我们先创建一个子节点head，是我们要插入的节点

	TreeNode head = new TreeNode(val);

3. 循环遍历二叉搜索树就可以了，终止条件是遍历到叶子结点

	TreeNode node  = root;
	while(node != null){
		if(val < node.val){
			//如果小于当前值，代表是左节点
			node = node.left;
		}else{
			//如果大于当前值，代表是右节点
            node = node.right;
        }
	}

4. 但有个问题，我们最后node一定是null，我们无法找到目标的根节点，我们使用parent记录根节点

	//记录父节点
	TreeNode parent = root;
	//遍历节点，给head找位置
	TreeNode node  = root;
	while(node != null){
		//记录当前父节点
		parent = node;
           
		if(val < node.val){
			//如果小于当前值，代表是左节点
            node = node.left;
		}else{
			//如果大于当前值，代表是右节点
            node = node.right;
		}
	}

5. 最后将节点插入，如何判断是左节点还是右节点
   • 因为 左树 < 根 < 右树
   • val < parent.val 那么就是左节点
   • val > parent.val 那么就是右节点

	if(val < parent.val){
		//小于父节点  是左节点
        parent.left = head;
	}else{
		//大于父节点  是右节点
		parent.right = head;
	}

6. 最后返回root

完整代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    //数据结构 第 13 天 2/2  树
    public TreeNode insertIntoBST(TreeNode root, int val) {
        TreeNode head = new TreeNode(val);
        if(root == null){
            return head;
        }
        //记录父节点
        TreeNode parent = root;
        //遍历节点，给head找位置
        TreeNode node  = root;
        while(node != null){
            //记录当前父节点
            parent = node;
           
            if(val < node.val){
                //如果小于当前值，代表是左节点
                node = node.left;
            }else{
                //如果大于当前值，代表是右节点
                node = node.right;
            }
        }

        if(val < parent.val){
            //小于父节点  是左节点
            parent.left = head;
        }else{
            //大于父节点  是右节点
            parent.right = head;
        }
        return root;
    }
}