235. 二叉搜索树的最近公共祖先
思路
二叉搜索树的查找最近公共祖先操作比普通二叉树容易很多,只需要判断两个结点位置如何。如果两节点值都比当前节点小,则向左遍历;如果两节点值都比当前节点大,则向右遍历;否则两节点位置则在当前节点两侧,此时当前节点就为最近公共祖先。
代码实现
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)return lowestCommonAncestor(root->left,p,q);
else if(root->val<p->val&&root->val<q->val)return lowestCommonAncestor(root->right,p,q);
else return root;
}
};
701.二叉搜索树中的插入操作
思路
首先还是递归老三样:
- 传入参数(略)
- 递归终止条件
遇到空节点则可以插入新叶子结点,然后返回新节点,让新节点被上一个结点接收 - 递归条件
当插入值小于当前结点值则向左递归
当插入值大于当前结点值则向右递归
代码实现
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(root==nullptr){
TreeNode* cur=new TreeNode(val);
return cur;
}
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&&root->right==nullptr){
delete root;
return nullptr;
}
else if(root->left&&root->right==nullptr){
TreeNode* cur=root->left;
delete root;
return cur;
}
else if(root->right&&root->left==nullptr){
TreeNode* cur=root->right;
delete root;
return cur;
}
else{
TreeNode *tmp=root;
TreeNode* cur=root->right;
while(cur->left)cur=cur->left;
cur->left=root->left;
tmp=root->right;
delete root;
return tmp;
}
}
if(root->val>key)root->left=deleteNode(root->left,key);
if(root->val<key)root->right=deleteNode(root->right,key);
return root;
}
};
总结
今天的题目写的越来越熟练啦,加油加油加油!