leetcode 211. Add and Search Word - Data structure design Trie树

题目链接

写一个数据结构， 支持两种操作。 加入一个字符串， 查找一个字符串是否存在。查找的时候， '.'可以代表任意一个字符。

显然是Trie树， 添加就是正常的添加， 查找的时候只要dfs查找就可以。 具体dfs方法看代码。

struct node
{
node *next[26];
int idEnd;
node() {
memset(next, NULL, sizeof(next));
idEnd = 0;
}
};
class WordDictionary {
public:
// Adds a word into the data structure.
node *root = new node();
void addWord(string word) {
node *p = root;
int len = word.size();
for(int i = 0; i<len; i++) {
if(p->next[word[i]-'a'] == NULL)
p->next[word[i]-'a'] = new node();
p = p->next[word[i]-'a'];
}
p->idEnd = 1;
return ;
}
int dfs(string word, int pos, node *p) {            //pos是代表当前是在哪一位。
if(pos == word.size())
return p->idEnd;
int len = word.size();
for(int i = pos; i<len; i++) {
if(word[i] == '.') {
for(int j = 0; j<26; j++) {
if(p->next[j] == NULL)
continue;
if(dfs(word, pos+1, p->next[j]))
return 1;
}
return 0;
} else if(p->next[word[i]-'a'] != NULL) {
return dfs(word, pos+1, p->next[word[i]-'a']);
} else {
return 0;
}
}
}
// Returns if the word is in the data structure. A word could
// contain the dot character '.' to represent any one letter.
bool search(string word) {
return dfs(word, 0, root);
}
};