class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
//节点空
if (root == nullptr) return root;
//节点不空
if (root->val == key) {
//节点左子树和右子树都空
if (root->left == nullptr && root->right == nullptr) {
delete root;
return nullptr;
} else if (root->left == nullptr) { //节点左子树空,右子树不空
auto rroot = root->right;
delete root;
return rroot;
} else if (root->right == nullptr) { //节点右子树空,左子树不空
auto lroot = root->left;
delete root;
return lroot;
} else { //节点左右节点都不空
TreeNode* cur = root->right;
while(cur->left != nullptr) { //找到右子树的最左节点
cur = cur->left;
}
cur->left = root->left;
auto node = root;
root = root->right;
delete node;
return root;
}
}
if (root->val > key) root->left = deleteNode(root->left, key);
if (root->val < key) root->right = deleteNode(root->right, key);
return root;
}
};
时间复杂度:O(h)h为树的高度
思路:相较于BST的插入操作,删除更为复杂,有五种情况,但是大致框架都是一样,为有返回值的遍历边(用返回值构造树的某条边),所以时间复杂度为O(h)。
五种情况如下:
(1)没有删除的节点,则遍历到空节点直接返回。
(2)找到删除的节点,其左右儿子为空,则直接删除该节点,返回为空。
(3)找到删除的节点,其左空右不空,删除该节点,返回右节点(右节点代替了该节点)。
(4)找到删除的节点,其左不空右空,删除该节点,返回左节点(同理)。
(5)找到删除的节点,左右都不空,将左子树放到右子树的最左的节点上,删除该节点,返回右节点。
知识点:delete可要可不要,虽然不会报错但是加着最好,因为c++没有自动回收机制,而Java有。