Menu
CoddyTech

删除

Coddy 字典树:数据结构系列第 8 课 课程的第 8 课,共 14 课。

从 trie 中删除内容分为两步。简单的一步:找到该单词对应的终止节点,并将它的 isEndOfWord 设回 false。完成后,search 就不再报告这个单词,这正是用户关心的结果。

比较棘手的一步:剪除现在已无用的节点。如果某个子节点没有自己的子节点,而且它本身也不是其他单词的结尾,那么它就是无用的,因为今后的搜索或前缀遍历都无法到达它。我们通过向下递归找到这些节点,然后在返回时将它们剪除。

要注意两种情况。如果你尝试删除一个从未插入过的单词,就什么也不做。如果要删除的单词是另一个已存储单词的前缀(例如 trie 中也有 "card" 时删除 "car"),你必须清除标记,但保留所有节点,因为仍然可以通过 "card" 到达这些节点。

challenge icon

挑战

中等

在 Trie 类中添加一个 delete 方法。

它接收一个字符串 word:

  • 如果 trie 中存储了 word,取消标记其终止节点(isEndOfWord = false),并移除沿途因此成为孤立节点的节点(没有子节点,也不是其他单词的结尾)。
  • 如果 trie 中没有 word,则不做任何操作。

一种简洁的剪枝实现方式是使用递归辅助函数,它会返回一个布尔值,告诉调用者当前子节点是否可以移除。

亲手试试

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "trie.h"

int main() {
    Trie t;
    Trie_init(&t);
    char line[1024];
    while (fgets(line, sizeof(line), stdin)) {
        line[strcspn(line, "\r\n")] = '\0';
        char* cmd = strtok(line, " \t");
        if (!cmd) continue;
        if (strcmp(cmd, "rootIsEmpty") == 0) { printf("%s\n", TrieNode_childrenCount(t.root) == 0 ? "true" : "false"); }
        if (strcmp(cmd, "hasChild") == 0) { char* arg = strtok(NULL, " \t"); if (arg) printf("%s\n", t.root->children[(unsigned char)arg[0]] != NULL ? "true" : "false"); }
        if (strcmp(cmd, "insert") == 0) { char* arg = strtok(NULL, " \t"); if (arg) Trie_insert(&t, arg); }
        if (strcmp(cmd, "search") == 0) { char* arg = strtok(NULL, " \t"); if (arg) printf("%s\n", Trie_search(&t, arg) ? "true" : "false"); }
        if (strcmp(cmd, "startsWith") == 0) { char* arg = strtok(NULL, " \t"); if (arg) printf("%s\n", Trie_startsWith(&t, arg) ? "true" : "false"); }
        if (strcmp(cmd, "delete") == 0) { char* arg = strtok(NULL, " \t"); if (arg) Trie_delete(&t, arg); }
    }
    return 0;
}

字典树:数据结构系列第 8 课中的所有课

自己练习: C 在线编译器