提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
前言
今天是跟着代码随想录刷题的第22天,主要是做了235. 二叉搜索树的最近公共祖先、701.二叉搜索树中的插入操作、450.删除二叉搜索树中的节点。
链接地址:二叉搜索树的最近公共祖先、二叉搜索树中的插入操作、删除二叉搜索树中的节点
一、二叉搜索树的最近公共祖先
思路:如果当前节点比p和q都大,二叉搜索树在节点左边,如果当前节点比p和q都小,二叉搜索树在节点有点,如果当前节点在p和q中间,二叉搜索树就是当前节点。
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root==NULL) return NULL;
if(root->val>p->val&&root->val>q->val)
{
TreeNode* left=lowestCommonAncestor(root->left,p,q);//只有这一句他只会返回根节点
if(left!=NULL) return left;
}
if(root->val<p->val&&root->val<q->val)
{
TreeNode* right=lowestCommonAncestor(root->right,p,q);
if(right!=NULL) return right;
}
return root;
}
};
同样的,这种需要知道左右是不是存在的需要左右中来遍历。
二、二叉搜索树中的插入操作
思路:找到第一个他小于或者大于的一个空节点,然后在那里创立一个节点,然后把值给放进去。
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if (root == NULL) {
return new TreeNode(val);
}
if (root->val < val) {
root->right = insertIntoBST(root->right, val);
} else {
root->left = insertIntoBST(root->left, val);
}
return root;
}
};
三、删除二叉搜索树中的节点
思路:考虑五种情况,本身是空,左不空右空,左空右不空,左右都空,左右都不空,然后最后一种比较复杂,然后记得用root的左右节点来接,这样,返回的节点就能接上了。
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if(root==NULL) return NULL;
if(root->val==key)
{
if(root->left!=NULL&&root->right==NULL)
{
return root->left;
}
if(root->right!=NULL&&root->left==NULL)
{
return root->right;
}
if(root->left!=NULL&&root->right!=NULL)
{
TreeNode* cur=root->right;
while(cur->left!=NULL)
{
cur=cur->left;
}
cur->left=root->left;
return root->right;
}
if(root->left==NULL&&root->right==NULL)
{
return NULL;
}
}
root->left=deleteNode(root->left,key);
root->right=deleteNode(root->right,key);
return root;
}
};