开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第22天，点击查看活动详情

现在就是大逃杀

刷题的日常-2022年12月14号

一天一题，保持脑子清爽

二叉搜索树中的插入操作

来自leetcode的 701 题，题意如下：

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

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

理解题意

通过题意，我们可以将信息整理如下：

• 题目给出一个二叉树 和 一个值
• 要求我们将这个值放入二叉树当中
• 注意，这棵二叉树是搜索树，插入之后必须要符合二叉搜索树的规则，既 左子树所有节点的值小于当前节点，右子树所有节点的值要大于当前节点。

做题思路

题目要求返回的是任意满足条件的二叉搜索树，那么其实就可以不关心平衡问题。所以我们可以将小于的数直接往树的左子树放，大的就往右子树放。然后递归以上的过程，如果出现左子树或者右子树没有的情况，直接放进去即可。步骤如下：

• 如果根节点为空，直接返回插入的节点
• 如果当前节点大于 插入的值，并且左子树是空的，直接放进左子树，否则进行递归
• 如果当前节点小于 插入的值，并且右子树是空的，直接放进右子树，否则进行递归
• 最后返回根节点即可

代码实现

代码实现如下:

class Solution {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            return new TreeNode(val);
        }
        if (root.val > val) {
            if (root.left == null) {
                root.left = new TreeNode(val);
            } else {
                insertIntoBST(root.left, val);
            }
        }
        if (root.val < val) {
            if (root.right == null) {
                root.right = new TreeNode(val);
            } else {
                insertIntoBST(root.right, val);
            }
        }
        return root;
    }
}