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

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

题目

解法

1. 递归：利用二叉搜索树的特性【没看题解】

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if(p->val > q->val) {
            TreeNode* temp = q;
            q = p;
            p = temp; 
        }
        // 终止条件
        if(root->val <= q->val && root->val >= p->val ) return root;
        // 单层递归逻辑
        if(root->val < p->val) {
            TreeNode* left = lowestCommonAncestor(root->right, p, q);
            return left;
        }else {
            TreeNode* right = lowestCommonAncestor(root->left, p, q);
            return right;
        }
    }
};

2.迭代

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        while (root) {
            if (root->val > p->val && root->val > q->val) root = root->left;
            else if (root->val < p->val && root->val < q->val) root = root->right;
            else return root;
        }
        return NULL; // 所有控制路径上都要有返回值
    }
};

基础知识补充

二叉树的深度和高度

• 二叉树节点的深度：指从根节点到该节点的最长简单路径边的条数或者节点数（取决于深度从0开始还是从1开始）
• 二叉树节点的高度：指从该节点到叶子节点的最长简单路径边的条数或者节点数（取决于高度从0开始还是从1开始）

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

题目

解法

1.递归

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

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

题目

解法

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (root == nullptr) return root; // 第一种情况，没有找到值对应的节点;
        if (root->val == key) {
            if(root->left == nullptr && root->right == nullptr) {// 第二种情况，值对应的节点是叶子节点，删除节点就OK
                delete root;
                return nullptr;
            } else if (root->left == nullptr) { // 第三种情况 值对应的节点的左子树为空，右子树不为空，删除节点，把右子树根节点顶上
                TreeNode* temp = root->right;
                delete root;
                return temp;
            } else if (root->right == nullptr) {// 第四种情况 值对应的节点的左子树不为空，右子树为空，删除节点，把左子树根节点顶上
                TreeNode* temp = root->left;
                delete root;
                return temp;
            } else { // 第五种情况，左右子树都不为空，把左子树放到右子树最左叶节点的后面
                TreeNode* cur = root->right; //找到右子树最左边的节点
                while (cur->left) {
                    cur = cur->left;
                }
                cur->left = root->left;
                TreeNode* temp = root;
                root = root->right;
                delete temp;
                return root;
            }
        }
        if (key < root->val) root->left = deleteNode(root->left, key);
        if (key > root->val) root->right = deleteNode(root->right, key);
        return root;
    }
};

感悟

删除节点比添加节点难