Menu
Coddy logo textTech

Liczba słów

Lekcja 9 z 14 w kursie Drzewa Trie — struktury danych #8 w Coddy.

Ile różnych słów znajduje się teraz w tym trie? Nie ma licznika, który na bieżąco śledzi tę wartość, więc musimy przejść przez drzewo i policzyć węzły, dla których isEndOfWord ma wartość true.

Wyszukiwanie w głąb rozpoczynające się od korzenia wykona to zadanie za jednym przejściem. Dla każdego węzła dodaj 1, jeśli oznacza koniec słowa, a następnie rekurencyjnie przejdź do każdego dziecka i zsumuj wyniki. Suma to liczba zapisanych słów.

Zauważ, że w ten sposób duplikaty są obsługiwane naturalnie: trzykrotne wstawienie "cat" nadal oznacza jeden węzeł jako koniec słowa, więc liczba wynosi 1. Nie są też liczone prefiksy, które nigdy nie zostały wstawione jako osobne słowa — ich węzły istnieją, ale isEndOfWord ma dla nich wartość false.

challenge icon

Wyzwanie

Łatwy

Dodaj metodę wordCount do klasy Trie.

Nie przyjmuje żadnych danych wejściowych i zwraca liczbę różnych słów aktualnie przechowywanych w drzewie trie. Przejdź przez drzewo trie od korzenia i policz każdy węzeł, dla którego isEndOfWord ma wartość true.

Spróbuj swoich sił

#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;
}

Wszystkie lekcje w sekcji Drzewa Trie — struktury danych #8

Poćwicz samodzielnie: Kompilator C online