Day19:235. 二叉搜索树的最近公共祖先、701.二叉搜索树中的插入操作、450.删除二叉搜索树中的节点

---

235. 二叉搜索树的最近公共祖先

题目链接

思路

二叉搜索树的查找最近公共祖先操作比普通二叉树容易很多，只需要判断两个结点位置如何。如果两节点值都比当前节点小，则向左遍历；如果两节点值都比当前节点大，则向右遍历；否则两节点位置则在当前节点两侧，此时当前节点就为最近公共祖先。

代码实现

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(root==NULL)return NULL;
        if(root->val>p->val&&root->val>q->val)return lowestCommonAncestor(root->left,p,q);
        else if(root->val<p->val&&root->val<q->val)return lowestCommonAncestor(root->right,p,q);
        else return root;
    }
};

---

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

题目链接

思路

首先还是递归老三样：

1. 传入参数（略）
2. 递归终止条件
   遇到空节点则可以插入新叶子结点，然后返回新节点，让新节点被上一个结点接收
3. 递归条件
   当插入值小于当前结点值则向左递归
   当插入值大于当前结点值则向右递归

代码实现

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if(root==nullptr){
            TreeNode* cur=new TreeNode(val);
            return cur;
        }
        if(root->val>val)root->left=insertIntoBST(root->left,val);
        if(root->val<val)root->right=insertIntoBST(root->right,val);
        return root;
    }
};

---

450.删除二叉搜索树中的节点

题目链接

思路

1. 传入参数（略）
2. 递归终止条件
   • 没找到符合条件结点：返回空指针
   • 找到符合条件结点：
     1. 该节点为叶子结点：直接删除
     2. 该节点只有左子树或只有右子树：删除该节点后返回他的左子树或右子树
     3. 该节点既有左子树又有右子树：把该节点左子树插入到右子树最小结点，删除该节点再返回该节点的右子树
3. 递归条件
   当插入值小于当前结点值则向左递归
   当插入值大于当前结点值则向右递归
4. 返回值：返回根节点

代码实现

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root==nullptr)return nullptr;
        if(root->val==key){
            if(root->left==nullptr&&root->right==nullptr){
                delete root;
                return nullptr;
            }
            else if(root->left&&root->right==nullptr){
                TreeNode* cur=root->left;
                delete root;
                return cur;
            }
            else if(root->right&&root->left==nullptr){
                TreeNode* cur=root->right;
                delete root;
                return cur;
            }
            else{
                TreeNode *tmp=root;
                TreeNode* cur=root->right;
                while(cur->left)cur=cur->left;
                cur->left=root->left;
                tmp=root->right;
                delete root;
                return tmp;
            }
        }
        if(root->val>key)root->left=deleteNode(root->left,key);
        if(root->val<key)root->right=deleteNode(root->right,key);
        return root;
    }
};

---

总结

今天的题目写的越来越熟练啦，加油加油加油！