Menu
Coddy logo textTech

単語数 (WordCount)

Coddyの「トライ木 - データ構造シリーズ #8」コースのレッスン 9/14。

現在、このトライ(trie)には何個の異なる単語が含まれていますか?実行中のカウンターはないため、ツリーを走査して isEndOfWordtrue であるノードをカウントする必要があります。

ルートから開始する深さ優先探索により、1回のパスで処理できます。各ノードについて、それが単語の終端であれば1を加え、次にすべての子ノードに対して再帰的に処理を行い、その結果を合計します。その合計が、格納されている単語の数になります。

これにより、重複が自然に処理されることに注目してください。"cat" を3回挿入しても、1つのノードが単語の終端としてマークされるだけなので、カウントは1のままです。それ自体が挿入されたわけではない接頭辞もカウントされません。それらのノードは存在しますが、isEndOfWordfalse になっているからです。

challenge icon

チャレンジ

簡単

Trie クラスに wordCount メソッドを追加してください。

このメソッドは引数を取らず、現在トライ木に格納されている一意の単語の数を返します。ルートからトライ木を走査し、isEndOfWordtrue であるすべてのノードをカウントしてください。

自分で試してみよう

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

トライ木 - データ構造シリーズ #8のすべてのレッスン