删叶子并归还内存

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

补全 erase 里「只有一个孩子或没有孩子」那一支：接好孩子，再 drop 掉自己。删掉叶子 13 后，输出「剩几个没归还」和中序，格式如 剩4个:甲-乙-丙-丁。

（本题用 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) {
            // TODO: 先记下唯一的孩子（可能是空），再 drop(root)，最后把孩子返回
            return root;
        }
        BNode* m = root->right;
        while (m->left != nullptr) m = m->left;
        root->val = m->val;
        root->right = erase(root->right, m->val);
    }
    return root;
}

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

本次输入：

输出：

👩‍🏫

💬 题目评论

删叶子并归还内存

更多题目

全部评论