Trie树C语言实现

最新推荐文章于 2023-05-11 21:08:45 发布

Soft'Wind

最新推荐文章于 2023-05-11 21:08:45 发布

阅读量1.1k

点赞数 3

本文链接：https://blog.csdn.net/qq_18431031/article/details/109218045

版权

这段代码展示了如何用C++实现Trie树数据结构，用于判断单词是否存在以及是否以特定前缀开头。通过`trieCreate`、`trieInsert`、`trieSearch`和`trieStartsWith`等函数，实现了插入单词、搜索单词和检查前缀的功能。在主函数中，首先读取一系列单词插入Trie树，然后针对不同前缀进行判断并输出结果。

摘要由CSDN通过智能技术生成

实现了输入单词，判断单词是否存在以及是否存在该单词的前缀

#include <string.h>
#include <stdlib.h>
#include <stdio.h>

typedef struct trie {
    int isEnd;
    struct trie *next[26];
} Trie;

Trie * trieCreate()
{
    Trie *root = (Trie *)malloc(sizeof(Trie));
    memset(root, 0, sizeof(*root));
    root->isEnd = 0;
    return root;
}

void trieInsert(Trie *obj, char *word)
{
    Trie *node = obj;
    for (int i = 0; word[i] != '\0'; i++) {
        char c = word[i];
        if (node->next[c - 'a'] == NULL)
            node->next[c - 'a'] = trieCreate();
        node = node->next[c - 'a'];
    }
    node->isEnd = 1;
}

bool trieSearch(Trie *obj, char *word)
{
    Trie *node = obj;
    for (int i = 0; word[i] != '\0'; i++) {
        char c = word[i];
        if (node->next[c - 'a'] == NULL)
            return false;
        node = node->next[c - 'a'];
    }
    return node->isEnd > 0;
}

bool trieStartsWith(Trie *obj, char *prefix)
{
    Trie *node = obj;
    for (int i = 0; prefix[i] != '\0'; i++) {
        char c = prefix[i];
        if (node->next[c - 'a'] == NULL)
            return false;
        node = node->next[c - 'a'];
    }
    return true;
}

void trieFree(Trie *obj)
{
    if (obj == NULL)
        return;
    for (int i = 0; i < 26; i++) {
        if (obj->next[i])
            trieFree(obj->next[i]);
    }
    free(obj);
}

int main()
{
    Trie *root = trieCreate();
    char word[50];
    while (scanf("%s", word) != EOF) {
        trieInsert(root, word);
    }
    char a[50];
    while (scanf("%s", a) != EOF) {
        if (trieStartsWith(root, a))
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}