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

题目：

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

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

提示：

• 给定的树上的节点数介于 0 和 10^4 之间
• 每个节点都有一个唯一整数值，取值范围从 0 到 10^8
• -10^8 <= val <= 10^8
• 新值和原始二叉搜索树中的任意节点值都不同

题解：

可以不考虑题目中提示所说的改变树的结构的插入方式。按照二叉搜索树插入值的方式。

步骤：

1.确定递归函数参数以及返回值

参数：根节点指针，以及要插入元素

返回值可以有，也可以没有，但递归函数如果没有返回值的话，实现是比较麻烦的

有返回值的话，可以利用返回值完成新加入的节点与其父节点的赋值操作。（解法选择有返回值）

递归函数的返回类型为节点类型TreeNode。

2.确定终止条件

终止条件就是找到遍历的节点为null的时候，就是要插入节点的位置了，并把插入的节点返回。

3.确定单层递归的逻辑

此时要明确不需要遍历整棵树，搜索树是有方向的，可以根据插入元素的数值，决定递归方向。

通过递归函数返回值完成了新加入节点的父子关系赋值操作了，下一层将加入节点返回。

/**
 * 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 {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) { //找到插值位置
            root = new TreeNode(val);
            return root;
        }
        if (val < root.val) {
            root.left = insertIntoBST(root.left, val);
        }
        if (val > root.val) {
            root.right = insertIntoBST(root.right, val);
        }
        return root;
    }

    // public void traversal(TreeNode root, int val) {
    //     //这样其实没有把新创建的节点给上一个节点，所以需要返回值递归回去
    //     // if (root == null) {
    //     //     root = new TreeNode(val);
    //     //     return;
    //     // }

    //     if (val < root.val) {
    //         traversal(root.left, val);
    //     } else {
    //         traversal(root.right, val);
    //     }
    // }
}

参考：代码随想录