代码随想录算法训练营第二十二天|235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

最新推荐文章于 2024-10-01 23:05:47 发布

MoonBlue:)

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量292

点赞数

分类专栏：代码随想录训练营文章标签：算法数据结构 java

本文链接：https://blog.csdn.net/Hireath_/article/details/130755536

版权

代码随想录训练营专栏收录该内容

48 篇文章 1 订阅

订阅专栏

文章介绍了如何在二叉搜索树中找到最近公共祖先，以及进行插入和删除节点的算法。对于最近公共祖先，关键在于利用二叉搜索树的性质从上往下遍历。插入操作是基于树的有序性自然进行。删除节点时，需处理四种情况，包括左右子节点为空的情况和旋转操作。

摘要由CSDN通过智能技术生成

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

思路

所以当我们从上向下去递归遍历，第一次遇到 cur节点是数值在[p, q]区间中，那么cur就是 p和q的最近公共祖先。（与普通二叉树不同）

代码

class Solution {
public:
    TreeNode* travelsal(TreeNode* cur, TreeNode* p, TreeNode* q) {
        if(cur == NULL) return cur;

        if(cur->val > p->val && cur->val > q-> val) {
            TreeNode* left = travelsal(cur->left,p,q);
            if(left != NULL) return left;
        }
        else if(cur->val < p->val && cur->val < q->val) {
            TreeNode* right = travelsal(cur->right,p,q);
            if(right != NULL) return right;
        }
        return cur;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        return travelsal(root,p,q);
    }
};

总结

普通二叉树找公共父节点是葱底向上，二叉搜索树是从上向下，二叉搜索树自身的性质会帮助解题

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

思路

二叉搜索树的有序结构使插入操作的路径天然形成

代码

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;
    }
};

总结

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

思路

删除节点的难点在于当被删除节点的左右子节点都不为空时，要做一个旋转的操作

代码

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root == NULL) return root;
        if(root->val < key) root->right = deleteNode(root->right,key);
        else if(root->val > key) root->left = deleteNode(root->left,key);
        else {
            // 1. 左右孩子都为空，直接删除
            if(root->left == NULL && root->right == NULL) {
                delete root;
                return NULL;
            }
            // 2. 左孩子为空，右孩子补位
            if(root->left == NULL && root->right) {
                TreeNode* retNode = root->right;
                delete root;
                return retNode;
            }
            // 3. 右孩子为空，左孩子补位
            else if(root->right == NULL && root->left) {
                TreeNode* retNode = root->left;
                delete root;
                return retNode;
            }
            // 4. 左右都不为空
            else {
                TreeNode* cur = root->right;
                while(cur->left != NULL) {
                    cur = cur->left;
                }
                cur->left = root->left;
                TreeNode* tmp = root;
                root = root->right;
                delete tmp;
                return root;
            }
        }
        return root;
    }
};