力扣题解-701. 二叉搜索树中的插入操作

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

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

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

例如,
给定二叉搜索树:

        4
       / \
      2   7
     / \
    1   3

和 插入的值: 5
你可以返回这个二叉搜索树:

         4
       /   \
      2     7
     / \   /
    1   3 5

或者这个树也是有效的:

         5
       /   \
      2     7
     / \   
    1   3
         \
          4

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-into-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

直接将待插入结点插入到二查搜索树的叶子结点上。

方法为：根据待插入数值与当前结点的值进行比较，判断将插入到其左子树还是右子树。

不管是插入左子树还是右子树，只有当该结点的左子树或右子树为叶子结点时，才进行插入；

否则继续查找相应的左子树或右子树。

代码（迭代方法）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (!root) {
            return new TreeNode(val);
        }

        TreeNode* cur = root;
        while (cur) {
            if (val < cur->val) {//判断是否可插入左子树
                if (!cur->left) {//左子树到达叶子，插入
                    cur->left = new TreeNode(val);
                    break;
                } else {//否则继续找左子树的叶子
                    cur = cur->left;
                }
            } else {//插入右子树
                if (!cur->right) {//右子树达到叶子，插入
                    cur->right = new TreeNode(val);
                    break;
                } else {//否则继续找右子树的叶子
                    cur = cur->right;
                }
            }
        }
        return root;
    }
};

代码（递归方法）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (!root) {//达到叶子结点即插入
            return new TreeNode(val);
        }

        if (val < root->val) {
            root->left = insertIntoBST(root->left, val);//插入左子树
        } else {
            root->right = insertIntoBST(root->right, val);//插入右子树
        }
        return root;
    }
};