Implement a trie with insert, search, and startsWith methods.
Note:
You may assume that all inputs are consist of lowercase letters a-z.
s思路:
1. trie,前缀树。用来搜索string很方便快速。每个节点包括26个指针数组,对应26个字母,如果child[0]不为空,表示这个字母存在,否则没这个字符;还有是否是单词结尾的标志符。
2. 如何insert? 需要对单词从左往右dfs遍历,比如:”bat”,首先看trie根节点指向的26个child的child[1]是否存在(不存在,用NULL表示),存在就进入下一个层次,不存在则需要新建一个node,并让child[1]指向这个节点。
3. 如何搜索?搜索和insert很类似,都是通过dfs一层一层的往下找,某个位置如果没指针,表示没找到;最后位置如果没有isWord表示也没有。这里就显示单词结尾符号的用处了!
4. 如何startswith?比如:查找是否含有以ab开头的单词。也是搜索,不过不需要判断单词结尾即可!
struct node{
node* child[26];
bool isWord;
node(){
for(int i=0;i<26;i++)
child[i]=NULL;
isWord=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(int i=0;i<word.size();i++){
int idx=word[i]-'a';
if(!cur->child[idx]){//没有这个字母
cur->child[idx]=new node();
}
cur=cur->child[idx];
}
cur->isWord=true;
}
/** Returns if the word is in the trie. */
bool search(string word) {
node* cur=root;
for(int i=0;i<word.size();i++){
int idx=word[i]-'a';
if(!cur->child[idx]){//没有这个字母
return false;
}
cur=cur->child[idx];
}
return cur->isWord;
}
/** Returns if there is any word in the trie that starts with the given prefix. */
bool startsWith(string prefix) {
node* cur=root;
for(int i=0;i<prefix.size();i++){
int idx=prefix[i]-'a';
if(!cur->child[idx]){//没有这个字母
return false;
}
cur=cur->child[idx];
}
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);
*/