208. 实现 Trie (前缀树)

最新推荐文章于 2021-04-14 20:07:03 发布

小桥落花流水

最新推荐文章于 2021-04-14 20:07:03 发布

阅读量74

点赞数

分类专栏： leetcode 前缀树文章标签： 1024程序员节

本文链接：https://blog.csdn.net/weixin_40306139/article/details/109267721

版权

leetcode 同时被 2 个专栏收录

171 篇文章 2 订阅

订阅专栏

前缀树

1 篇文章 0 订阅

订阅专栏

本文介绍了如何实现一个前缀树数据结构，包括插入、搜索和检查前缀匹配三个操作。通过创建一个节点类并用C++编写Trie类，实现了对字符串的有效管理和查找。示例展示了插入和搜索字符串的用法，强调了在字符串结束时使用is_end标志来判断是否为完整单词。

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

208. 实现 Trie (前缀树)

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert("apple");
trie.search("apple"); // 返回 true
trie.search("app"); // 返回 false
trie.startsWith("app"); // 返回 true
trie.insert("app");
trie.search("app"); // 返回 true
说明:

你可以假设所有的输入都是由小写字母 a-z 构成的。
保证所有输入均为非空字符串。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/implement-trie-prefix-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

基本思路：前缀树，注意是在字符串结束后的下一个的is_end来确定是否是结束。

struct Node {
    bool is_end;
    Node *next[26];
    Node(){
        memset(next,0,sizeof(next));
        is_end=false;
    };
};
class Trie {
private:
    Node *root;
public:
    /** Initialize your data structure here. */
    Trie() {
        root=new Node();
    }
    
    /** Inserts a word into the trie. */
    void insert(string word) {
        Node *cur=root;
        for(auto ch:word){
            int index=ch-'a';
            if(cur->next[index]==nullptr){
                cur->next[index]=new Node();
            }
            cur=cur->next[index];
        }
        cur->is_end=true;
    }
    
    /** Returns if the word is in the trie. */
    bool search(string word) {
        Node *cur=root;
        for(auto ch:word){
            int index=ch-'a';
            if(cur!=nullptr){
                cur=cur->next[index];
            }
            else
                return false;
        }
        return cur!=nullptr&&cur->is_end;
    }
    
    /** Returns if there is any word in the trie that starts with the given prefix. */
    bool startsWith(string prefix) {
        Node *cur=root;
        for(auto ch:prefix){
            int index=ch-'a';
            if(cur){
                cur=cur->next[index];
            }
            else    
                return false;
        }
        return cur!=nullptr;
    }
};