C语言设计题7-实现前缀树

最新推荐文章于 2024-07-02 17:44:31 发布

文卿的藏剑山庄

最新推荐文章于 2024-07-02 17:44:31 发布

阅读量156

点赞数

分类专栏： C语言设计题 leetcode 文章标签： c语言 leetcode 算法

本文链接：https://blog.csdn.net/qq_37651325/article/details/129393586

版权

leetcode 同时被 2 个专栏收录

142 篇文章 1 订阅

订阅专栏

C语言设计题

6 篇文章 0 订阅

订阅专栏

题目

208. 实现 Trie (前缀树)

代码

typedef struct tagTrie{
    struct tagTrie *children[26];
    bool isEnd;
} Trie;

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

void trieInsert(Trie *obj, char *word) {
    int n = strlen(word);
    for (int i = 0; i < n ; i++) {
        int ch = word[i] - 'a';
        if (obj->children[ch] == NULL) {
            obj->children[ch] = trieCreate();
        }
        obj = obj->children[ch];
    }
    obj->isEnd = true;
}

bool trieSearch(Trie *obj, char *word) {
    int n = strlen(word);
    for (int i = 0; i < n; i++) {
        int ch = word[i] - 'a';
        if (obj->children[ch] == NULL) {
            return false;
        }
        obj = obj->children[ch];
    }
    return obj->isEnd;
}

bool trieStartsWith(Trie *obj, char *prefix) {
    int n = strlen(prefix);
    for (int i = 0; i < n; i++) {
        int ch = prefix[i] - 'a';
        if (obj->children[ch] == NULL) {
            return false;
        }
        obj = obj->children[ch];
    }
    return true;
}

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