從二叉樹中刪除節點

Question

抽象類的二叉樹。 我有三種刪除方法，第一種是用戶菜單，其他是遞歸的。 刪除一個節點后，我遇到了減少節點計數的問題。 請幫忙，我應該在哪里寫“r->count--”？

Node* Del(Node *t, Node *t0) {
    if (t->left != nullptr) {
        t->left = Del(t->left, t0);
        return t;
    }
    t0->o = t->o;
    Node *x = t->right;
    delete t;
    return x;
}
Node* deleteNodes(Node *r, Node *elem, bool &deleted) {
    bool del;
    if (r == nullptr) {
        deleted = false;
        return r;
    } 
    if (equal(elem->o, r->o) < 0) {
        r->left = deleteNodes(r->left, elem, del);
        deleted = del;
        return r;
    }
    else if (equal(elem->o, r->o) > 0) {
        r->right = deleteNodes(r->right, elem, del);
        deleted = del;
        return r;
    }
    deleted = true;
    if (r->left == nullptr && r->right == nullptr) {
        delete r;
        return nullptr;
    }
    if (r->left == nullptr) {
        Node *x = r->right;
        delete r;
        return x;
    }
    if (r->right == nullptr) {
        Node *x = r->left;
        delete r;
        return x;
    }
    r->right = Del(r->right, r);
    return r;
}

Answer 1

您應該將它放在找到要刪除的節點的代碼中。 這也是您將deleted設置為true地方：

所以：

deleted = true;
r->count--;