一、235. 二叉搜索树的最近公共祖先
题目链接/文章讲解:代码随想录
视频讲解:235. 二叉搜索树的最近公共祖先
思考:关键在如何利用二叉搜索树的特性去找最近公共祖先
代码实现:
class Solution {
private:
TreeNode* traversal(TreeNode* cur, TreeNode* p, TreeNode* q) {
if (cur == NULL) return cur;
// 中
if (cur->val > p->val && cur->val > q->val) { // 左
TreeNode* left = traversal(cur->left, p, q);
if (left != NULL) {
return left;
}
}
if (cur->val < p->val && cur->val < q->val) { // 右
TreeNode* right = traversal(cur->right, p, q);
if (right != NULL) {
return right;
}
}
return cur;
}
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
return traversal(root, p, q);
}
};
精简版:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
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包含了root->val >p <q 和 <p >q 2种情况
} else return root;
}
};
迭代法:
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
while(root) {
if (root->val > p->val && root->val > q->val) {
root = root->left;
} else if (root->val < p->val && root->val < q->val) {
root = root->right;
} else return root;
}
return NULL;
}
};
二、701.二叉搜索树中的插入操作
题目链接/文章讲解:代码随想录
思考:代码简化的关键在如何通过递归函数返回值完成了新加入节点的父子关系赋值操作。即下一层将加入节点返回,本层用root->left或者root->right将其接住。
代码实现:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root,int val) {
if(root == NULL)
{
TreeNode* node = new TreeNode(val);
return node;
}
if(root->val > val )
{
root->left = insertIntoBST(root->left,val);
}
else if(root->val < val)
{
root->right = insertIntoBST(root->right,val);
}
return root;
}
};
三、450.删除二叉搜索树中的节点
题目链接/文章讲解:代码随想录
思考:找到目标节点就要进行删除操作,因此删除操作就在终止条件里。
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
最复杂的是第五种情况(以返回删除节点右孩子为新的根节点为例):
找到要删除的节点root后,要找比该节点稍稍大的节点的最左侧cur,执行root->left = cur->left.
代码实现:
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if(root == NULL) return root;
if(root->val == key)
{
if(root->left == NULL && root->right == NULL)
{
delete root;
return NULL;
}
else if(root->left == NULL)
{
auto retNode = root->right;
delete root;
return retNode;
}
else if(root->right == nullptr) {
auto retNode = root->left;
delete root;
return retNode;
}
else
{
TreeNode* cur = root->right;
while(cur->left != NULL)
{
cur = cur->left;
}
cur->left = root->left;
TreeNode* tmp = root;
root = root->right;
delete tmp;
return root;
}
}
if (root->val > key) root->left = deleteNode(root->left, key);
if (root->val < key) root->right = deleteNode(root->right, key);
return root;
}
};