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

题目

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

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

思路

• 确定递归的参数和返回值：参数就是根节点指针，以及要插入的元素，递归函数有返回值，可以利用返回值完成新加入的节点与其父节点的赋值操作
• 确定终止条件：当前便利的节点是NULL，则该节点就是要插入节点的位置，并把插入的节点返回
• 确定单层递归的逻辑：搜索树是有方向的，可以根据插入元素的数值决定递归的方向

代码

/**
 * 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 == NULL)
        {
            // 递归出口
            TreeNode* node = new TreeNode(val);// 新建一个节点
            return node;
        }
        if(root->val > val)
        {
            root->left = insertIntoBST(root->left,val);
        }
        if(root->val < val)
        {
            root->right = insertIntoBST(root->right,val);
        }
        return root;
    }
};

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32