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

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

二叉搜索树的性质：对于任意节点 \(root\) 而言，左子树（如果存在）上所有节点的值均小于 \(root.val\)，右子树（如果存在）上所有节点的值均大于 \(root.val\)，且它们都是二叉搜索树。

因此，当将 \(val\) 插入到以 \(root\) 为根的子树上时，根据 \(val\) 与 \(root.val\) 的大小关系，就可以确定要将 \(val\) 插入到哪个子树中。

• 如果该子树不为空，则问题转化成了将 \(val\) 插入到对应子树上。
• 否则，在此处新建一个以 \(val\) 为值的节点，并链接到其父节点 \(root\) 上。

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