题目

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

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

思路

根据二叉搜索树的特点,不断比较节点的值递归查找左右子树,直到找到空节点插入元素就可以了

  • 如果val比当前节点值小,继续递归左子树
  • 如果val比当前节点值大,继续递归右子树
  • 当递归当前节点为空,创建新Node返回
  • 递归函数返回值完成了新加入节点的父子关系赋值操作,下一层将加入节点返回,本层用root->left或者root->right将其接住

    1. class Solution
    2. {
    3. public:
    4. TreeNode *insertIntoBST(TreeNode *root, int val)
    5. {
    6. if (root == nullptr)
    7. return new TreeNode(val);
    8. if (root->val > val)
    9. root->left = insertIntoBST(root->left, val);
    10. else
    11. root->right = insertIntoBST(root->right, val);
    12. return root;
    13. }
    14. };