删除结点有三种情况:
1、待删除结点为叶子节点,不存在左右子。
直接删掉就行,返回nullptr回去就是删掉了。
2、待删除结点存在一个子结点
直接将子结点填补到待删除结点的位置即可,即直接返回子结点
3、待删除结点左右子都存在
则有两种删除方法:
》将其右子填补到待删除结点的位置,再将其左子接到右子的右子树中最左子的下方
》将其右子接到左子的左子树的最右处,再将左子填补到待删除结点的位置
先贴上我一遍过的代码🤤 再说问题
class Solution {
public:
TreeNode* pre = nullptr;//用pre指向前一个结点, 以帮助删除结点
TreeNode* deleteNode(TreeNode* root, int key) {
if (root == nullptr) return nullptr;
if (root->val == key) {//当前结点root就是要删除的结点
if (pre == nullptr) pre = root;
//第一种情况
if (!root->left && !root->right) return nullptr;
//第二种情况
else if (!root->left || !root->right) {
return root->left != nullptr ? root->left : root->right;
}
//第三种情况
else {
TreeNode* replace = root->right;
//先将替补填上去
if (key < pre->val) pre->left = root->right;
else if (key > pre->val) pre->right = root->right;
//再将原结点的另一个子树接到替补结点的最左子的左边
//》先遍历到替补结点的最左子node
TreeNode* node = replace;
while(node->left) node = node->left;
//》再将原结点的另一个子树接到最左子node的左边
node->left = root->left;
return replace;
}
}
pre = root;
if (key < root->val) root->left = deleteNode(root->left, key);
else if (key > root->val) root->right = deleteNode(root->right, key);
return root;
}
};
问题就是没有释放内存,因此在删除结点时都要记得添上这两句
delete root;
root==nullptr;