Contagem de Palavras
Lição 9 de 14 do curso Tries - Série de Estruturas de Dados #8 da Coddy.
Quantas palavras distintas existem nesta trie agora? Não há um contador em execução, então temos que percorrer a árvore e contar os nós onde isEndOfWord é true.
Uma busca em profundidade (depth-first search) começando pela raiz faz o trabalho em uma única passagem. Para cada nó, adicione 1 se ele for o fim de uma palavra, então faça a recursão em cada filho e some os resultados. O total é o número de palavras armazenadas.
Observe que isso lida naturalmente com duplicatas: inserir "cat" três vezes ainda marca apenas um nó como fim de palavra, então a contagem é 1. Prefixos que nunca foram inseridos por si mesmos também não são contados - seus nós existem, mas isEndOfWord é false neles.
Desafio
FácilAdicione um método wordCount à classe Trie.
Ele não recebe entrada e retorna o número de palavras distintas atualmente armazenadas na trie. Percorra a trie a partir da raiz e conte cada nó cujo isEndOfWord seja true.
Experimente você mesmo
#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;
}