leetcode刷题：二叉树26（二叉搜索树中的插入操作）

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

力扣题目链接

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

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

提示：

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

很简单的一题，按二叉搜索树的方式遍历，遇到叶子结点就直接判断大小，根据情况添加左或右；遇到单出度结点，先判断大小，如果是无出度的分支，则直接在本节点上添加新左或右结点，如果是有出度的分支，循环到那个结点再处理。

package com.programmercarl.tree;

/**
 * @ClassName InsertIntoBST
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/6 14:47
 * @Version 1.0
 * https://leetcode.cn/problems/insert-into-a-binary-search-tree/
 * 701. 二叉搜索树中的插入操作
 **/
public class InsertIntoBST {
    public TreeNode insertIntoBST(TreeNode root, int val) {
        //空根结点，就创建一个结点返回
        if (root == null) {
            return new TreeNode(val);
        }
        TreeNode node = root;
        insert(node, val);
        return root;
    }

    public void insert(TreeNode root, int val) {
        //当前结点是叶子结点
        if (root.left == null && root.right == null) {
            if (root.val > val) {
                //当前值比目标值大
                root.left = new TreeNode(val);
            } else {
                root.right = new TreeNode(val);
            }
            return;
        }
        //单出度情况
        if (root.left == null && root.right != null) {
            if (root.val > val) {
                //当前值比目标值大
                root.left = new TreeNode(val);
                return;
            }
        }

        if (root.left != null && root.right == null) {
            if (root.val < val) {
                //当前值比目标值小
                root.right = new TreeNode(val);
                return;
            }
        }
        if (root.val > val) {
            //当前值比目标值大
            insert(root.left, val);
        } else {
            insert(root.right, val);
        }
    }
}