代码随想录算法训练营 | 二叉树part07

最新推荐文章于 2024-09-29 16:46:50 发布

Fern_v

最新推荐文章于 2024-09-29 16:46:50 发布

阅读量170

点赞数 3

分类专栏： leetcode 文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/Fern_v/article/details/140737087

版权

leetcode 专栏收录该内容

35 篇文章 0 订阅

订阅专栏

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

可以使用 236. 二叉树的最近公共祖先的普通解法；
但考虑到二叉搜索树有序，判断当前节点值与p,q节点值的大小来判断p,q在那颗子树上，在决定是否遍历子树

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root->val > p->val && root->val > q->val) {  // p, q 均在左子树
            return lowestCommonAncestor(root->left, p, q);
        } else if (root->val < p->val && root->val < q->val) {  // p, q 均在右子树
            return lowestCommonAncestor(root->right, p, q);
        } 
        // else if (root->val > p->val && root->val < q->val || 
        //            root->val < p->val && root->val > q->val) {  // p, q 分别在左右子树
        //     return root;
        // } else if (root == p|| root == q) {
        //     return root;
        // }
        // return nullptr;
        return root;
    }
};

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

class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if(root == nullptr) {
            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 == nullptr) {
            return nullptr;
        }
        if (root->val == key) {
            if (root->left == nullptr) {
                return root->right;
            } else if (root->right == nullptr) {
                return root->left;
            } else {
                TreeNode* node = root->right;
                while(node->left) { // 寻找右子树的最左节点，即右子树上最小的值，大于左子树上的所有值
                    node = node->left;
                }
                node->left = root->left; 
                root = root->right;
                
            }
        } else if(root->val > key) { // 要删除的节点在左子树上
            root->left = deleteNode(root->left, key);
        } else if(root->val < key) { // 要删除的节点在右子树上
            root->right = deleteNode(root->right, key);
        }
        return root;
    }
};