算法训练Day19 | 235. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

WIIIter

已于 2024-05-08 21:29:38 修改

阅读量170

点赞数 5

分类专栏： LeetCode刷题文章标签：算法 leetcode c++ 数据结构

于 2024-03-25 22:37:00 首次发布

本文链接：https://blog.csdn.net/WIIIter/article/details/137028171

版权

LeetCode刷题专栏收录该内容

49 篇文章 0 订阅

订阅专栏

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

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

参考文章：代码随想录-235. 二叉搜索树的最近公共祖先

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

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        TreeNode* node = new TreeNode(val);
        TreeNode* temp = root;
        if(!temp) return node;
        while(temp){
            if(temp->val > val){
                if(temp->left) temp = temp->left;
                else{
                    temp->left = node;
                    break;
                } 
                
            }
            else if(temp->val < val){
                if(temp->right) temp = temp->right;
                else{
                    temp->right = node;
                    break;
                } 
            }
            else return nullptr;
        }
        return root;
    }
};

参考文章：代码随想录-701.二叉搜索树中的插入操作

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

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root == nullptr){
            return root;
        }
        if(root->val == key){
            if(root->right == nullptr){
                return root->left;
            }
            TreeNode* tmp = root->right;
            while(tmp->left){
                tmp = tmp->left;
            }
            swap(root->val, tmp->val);
        }
        root->left = deleteNode(root->left, key);
        root->right = deleteNode(root->right, key);

        return root;


    }
};