删双孩子并归还内存

👁️ 1 人浏览 💬 0 人评论 ❤️ 添加收藏

同一个 erase，这次删 17（左右都有孩子）。补全双孩子那一支：抄后继的值，再去右子树里把后继删掉。输出格式同上。

（本题用 g++ -std=c++17 -O0 编译。）

提交你的答案

请登录后提交答案。

去登录

← 删单孩子并归还内存整棵树全部释放 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

#include <iostream>
#include <string>
#include <vector>
using namespace std;

struct BNode {
    int val;
    BNode* left = nullptr;
    BNode* right = nullptr;
};

int alive = 0;   // 还有几个节点没有 delete

BNode* make(int v) {
    alive++;
    return new BNode{v};
}

void drop(BNode* nd) {
    alive--;
    delete nd;
}

BNode* insert(BNode* root, int val) {
    if (root == nullptr) return make(val);
    if (val < root->val) root->left = insert(root->left, val);
    else if (val > root->val) root->right = insert(root->right, val);
    return root;
}

BNode* build() {
    BNode* root = nullptr;
    for (int v : {17, 24, 15, 13, 23}) root = insert(root, v);
    return root;
}

void mid(BNode* node, vector<int>& out) {
    if (node == nullptr) return;
    mid(node->left, out);
    out.push_back(node->val);
    mid(node->right, out);
}

BNode* erase(BNode* root, int val) {
    if (root == nullptr) return nullptr;
    if (val < root->val) {
        root->left = erase(root->left, val);
    } else if (val > root->val) {
        root->right = erase(root->right, val);
    } else {
        if (root->left == nullptr || root->right == nullptr) {
            BNode* child = root->left ? root->left : root->right;
            drop(root);
            return child;
        }
        BNode* m = root->right;
        while (m->left != nullptr) m = m->left;
        // TODO: 把 m 的值抄到 root 上，再去 root 的右子树里把那个值删掉（它会被 drop）
    }
    return root;
}

int main() {
    BNode* root = erase(build(), 17);
    vector<int> out;
    mid(root, out);
    cout << "剩" << alive << "个:";
    for (size_t i = 0; i < out.size(); i++) cout << (i ? "-" : "") << out[i];
    cout << endl;
}

本次输入：

输出：

👩‍🏫

💬 题目评论

删双孩子并归还内存

更多题目

全部评论