代码随想录Day22——35. 二叉搜索树的最近公共祖先 701.二叉搜索树中的插入操作 450.删除二叉搜索树中的节点

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

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

class Solution {
public:
    TreeNode* get(TreeNode* root, TreeNode* p, TreeNode* q)
    {
        TreeNode* left = NULL;
        TreeNode* right = NULL;
        if(root == NULL) return NULL;
        left = get(root->left,p,q);
        right = get(root->right,p,q);
        if(root == q || root == p) return root;
        if(left && right) return root;
        if((left != NULL && right == NULL)) return left;
        if((left == NULL && right != NULL)) return right;
        return NULL;
    }
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        return get(root,p,q);
    }
};
  • 使用的方法是适用于普通二叉树的,将昨天没写出来的代码写了一边,对于二叉搜索树有更简单的思路:即找从上到下第一次出现介于pq两个数之间的节点就是最近公共祖先:代码随想录 (programmercarl.com)

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

给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。

class Solution {
private:
    TreeNode* parent;
    void traversal(TreeNode* cur, int val) {
        if (cur == NULL) {
            TreeNode* node = new TreeNode(val);
            if (val > parent->val) parent->right = node;
            else parent->left = node;
            return;
        }
        parent = cur;
        if (cur->val > val) traversal(cur->left, val);
        if (cur->val < val) traversal(cur->right, val);
        return;
    }

public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        parent = new TreeNode(0);
        if (root == NULL) {
            root = new TreeNode(val);
        }
        traversal(root, val);
        return root;
    }
};
  • 实际思路很简单,从上往下找,如果节点大于插入值就往左走,小于插入值就往右走,直到找到叶子节点,将插入值插入叶子节点下。
  • 一开始陷入了思维误区,使用双指针找两个节点使插入值介于两个节点之间,但这是错误的,因为左子树上可能会有比插入值大的情况。

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

给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。

/**
 * 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* get(TreeNode* root,int key)
    {
        if(root == NULL) return NULL;
        if(root->val == key)
        {
            if(root->left == NULL && root->right == NULL)
            {
                return NULL;
            }
            else if(root->left != NULL && root->right == NULL)
            {
                return root->left;
            }
            else if(root->left == NULL && root->right != NULL)
            {
                return root->right;
            }
            else
            {
                TreeNode* temp = root->right;
                while(temp != NULL && temp->left != NULL)
                {
                    temp = temp->left;
                }
                temp->left = root->left;
                return root->right;
            }
        }
        root->left = get(root->left,key);
        root->right = get(root->right,key);
        return root;
    }
    TreeNode* deleteNode(TreeNode* root, int key) {
        return get(root,key);
    }
};
  • 自己没思路,下午看的讲解晚上自己敲出来了。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值