二十天继续打卡。插入操作不用改变树的结构,删除操作比较麻烦
235.二叉搜索树的最近公共祖先
做题过程
- 用昨天的二叉树的最近公共祖先思路做出来,没用二叉搜索树的性质。
- 本题可用二叉搜索树的性质,采用迭代法返回第一个值在pq之前的节点返回
迭代法
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 root;
}
};
701.二叉搜索树中的插入操作
做题过程
- 采用迭代法做,寻找应加入的位置。
- 用递归法做,沿着一条路径修改。
迭代法
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
TreeNode* node = new TreeNode(val);
if (root == nullptr) return node;
TreeNode* cur = root;
TreeNode* pre;
while (cur) {
pre = cur;
if (cur->val > val) cur = cur->left;
else cur = cur->right;
}
if (pre->val > val) pre->left = node;
else pre->right = node;
return root;
}
};
450.删除二叉搜索树中的节点
做题过程
- 没想出怎么处理删除节点的逻辑
知识点
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
递归法、
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if (root == nullptr) return nullptr;
if (root->val < key) root->right = deleteNode(root->right, key);
else if (root->val > key) root->left = deleteNode(root->left, key);
else {
if (!root->left && !root->right) {
delete root;
return nullptr;
} else if (root->left && !root->right) {
TreeNode* temp = root->left;
delete root;
return temp;
} else if (!root->left && root->right) {
TreeNode* temp = root->right;
delete root;
return temp;
} else {
TreeNode* cur = root->right;
while (cur->left) cur = cur->left;
cur->left = root->left;
TreeNode* temp = root->right;
delete root;
return temp;
}
}
return root;
}
};