LeetCode 208 Implement Trie(实现字典树)

Implement a trie with insertsearch, and startsWith methods.

Note:
You may assume that all inputs are consist of lowercase letters a-z.

题目大意:实现Trie树的insert、search和startsWith方法。

解题思路:关键在于对Trie树的理解。在我的实现里,结构体TrieNode没有保存任何字符,因为我觉得如果某个节点的指针不为空,那么就说明代表该节点的字符存在。

代码如下:

#define MAXN 27
typedef struct TrieNode{
    struct TrieNode* next[MAXN];
} Trie;

/** Initialize your data structure here. */
Trie* trieCreate() {
    Trie* root = malloc(sizeof(Trie));
    for(int i = 0;i < MAXN;i++){
        root->next[i] = NULL;
    }
    return root;
}

/** Inserts a word into the trie. */
void trieInsert(Trie* obj, char* word) {
    if(strlen(word) == 0){
        obj->next[26] = trieCreate();
        return ;
    }
    if(obj->next[*word - 'a'] == NULL){
        obj->next[*word - 'a'] = trieCreate();
        trieInsert(obj->next[*word - 'a'], word + 1);
    }else{
        trieInsert(obj->next[*word - 'a'], word + 1);
    }
}

/** Returns if the word is in the trie. */
bool trieSearch(Trie* obj, char* word) {
    if(strlen(word) == 0){
        if(obj->next[26]) return true;
        else              return false;
    }
    if(obj->next[*word - 'a']) return trieSearch(obj->next[*word - 'a'], word + 1);
    else                       return false;
}

/** Returns if there is any word in the trie that starts with the given prefix. */
bool trieStartsWith(Trie* obj, char* prefix) {
    if(strlen(prefix) == 1){
        if(obj->next[*prefix - 'a']) return true;
        else                         return false;
    }
    if(obj->next[*prefix - 'a']) return trieStartsWith(obj->next[*prefix - 'a'], prefix + 1);
    else                         return false;
}

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

/**
 * Your Trie struct will be instantiated and called as such:
 * struct Trie* obj = trieCreate();
 * trieInsert(obj, word);
 * bool param_2 = trieSearch(obj, word);
 * bool param_3 = trieStartsWith(obj, prefix);
 * trieFree(obj);
 */

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值