前缀树
应用
- 前缀匹配
- 词频统计
数据结果
- root节点
root = Trie()
- 孩子节点
HashMap<char,Trie> children
- 结束FLag
bool isEnd/isword
判断是否是数字 - 对应的值
string val
题目推荐
- leetcode 208
- leetcode 720
- leetcode 692
208. 实现 Trie (前缀树)
难度中等923
Trie(发音类似 “try”)或者说 前缀树 是一种树形数据结构,用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景,例如自动补完和拼写检查。
请你实现 Trie 类:
Trie()
初始化前缀树对象。void insert(String word)
向前缀树中插入字符串word
。boolean search(String word)
如果字符串word
在前缀树中,返回true
(即,在检索之前已经插入);否则,返回false
。boolean startsWith(String prefix)
如果之前已经插入的字符串word
的前缀之一为prefix
,返回true
;否则,返回false
。
示例:
输入
["Trie", "insert", "search", "search", "startsWith", "insert", "search"]
[[], ["apple"], ["apple"], ["app"], ["app"], ["app"], ["app"]]
输出
[null, null, true, false, true, null, true]
解释
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
class Trie {
public:
struct TrieNode {
map<char,TrieNode*> children;
bool isword; // 是否是一个单词
string word; // 不用遍历记录,直接返回单词
TrieNode():isword(false){}
};
TrieNode* root;
/** Initialize your data structure here. */
Trie() {
root = new TrieNode();
}
/** Inserts a word into the trie. */
void insert(string word) {
// 插入单词
TrieNode* index = root;
for(int i=0;i<word.size();i++) {
if(index->children.count(word[i])==0) {
index->children[word[i]]=new TrieNode();
}
index=index->children[word[i]];
}
// 结束以后
index->isword = true;
index->word = word;
}
/** Returns if the word is in the trie. */
bool search(string word) {
TrieNode* index = root;
for(int i=0;i<word.size();i++) {
if(index->children.count(word[i])==0) {
return false;
}
index=index->children[word[i]];
}
if(index->isword) {
return true;
}
return false;
}
/** Returns if there is any word in the trie that starts with the given prefix. */
bool startsWith(string prefix) { // 搜索前缀
TrieNode* index = root;
for(int i=0;i<prefix.size();i++) {
if(index->children.count(prefix[i])==0) {
return false;
}
index=index->children[prefix[i]];
}
return true;
}
};
/**
* Your Trie object will be instantiated and called as such:
* Trie* obj = new Trie();
* obj->insert(word);
* bool param_2 = obj->search(word);
* bool param_3 = obj->startsWith(prefix);
*/