WordCount
Урок 9 из 14 курса Префиксные деревья (Trie) — Серия «Структуры данных» №8 на Coddy.
Сколько уникальных слов находится в этом префиксном дереве прямо сейчас? Здесь нет текущего счетчика, поэтому нам нужно обойти дерево и посчитать узлы, где isEndOfWord имеет значение true.
Поиск в глубину, начинающийся с корня, выполняет эту задачу за один проход. Для каждого узла прибавьте 1, если он является концом слова, затем рекурсивно перейдите в каждый дочерний узел и суммируйте результаты. Итоговая сумма — это количество сохраненных слов.
Обратите внимание, что это естественным образом обрабатывает дубликаты: вставка "cat" трижды по-прежнему помечает только один узел как конец слова, поэтому счетчик будет равен 1. Префиксы, которые никогда не вставлялись как самостоятельные слова, также не учитываются — их узлы существуют, но isEndOfWord для них имеет значение false.
Задание
ЛегкоДобавьте метод wordCount в класс Trie.
Он не принимает никаких аргументов и возвращает количество уникальных слов, хранящихся в данный момент в префиксном дереве (trie). Обойдите дерево от корня и посчитайте каждый узел, у которого свойство isEndOfWord имеет значение true.
Попробуйте сами
#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); }
if (strcmp(cmd, "wordCount") == 0) { printf("%d\n", Trie_wordCount(&t)); }
}
return 0;
}