Menu
Coddy logo textTech

StartsWith

Урок 7 из 14 курса Префиксные деревья (Trie) — Серия «Структуры данных» №8 на Coddy.

Основная причина существования префиксных деревьев (trie) — это быстрые ответы на вопросы о префиксах. startsWith проверяет, начинается ли какое-либо сохраненное слово с заданного префикса — ему не важно, является ли сам этот префикс полным вставленным словом.

Обход почти идентичен методу search: начиная от корня, переходите к дочерним узлам по одному символу за раз. Если по пути символ отсутствует, значит, ни одно сохраненное слово не начинается с этого префикса, поэтому возвращаем false. Если вы прошли через все символы без остановки, ответ — true — поддерево под этим узлом содержит как минимум одно сохраненное слово.

Единственное отличие от search: в конце нет проверки isEndOfWord. Достаточно просто дойти до конца префикса.

challenge icon

Задание

Легко

Добавьте метод startsWith в класс Trie.

Он принимает строку prefix и возвращает:

  • true, если какое-либо из вставленных слов начинается с prefix (префикс может совпадать с полным словом или быть начальной частью более длинного слова).
  • false в противном случае.

Попробуйте сами

#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"); }
    }
    return 0;
}

Все уроки раздела Префиксные деревья (Trie) — Серия «Структуры данных» №8