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); } };