Trie 字典树模板
#include <cstring> #include <vector> #include <cstdio> using namespace std; //*********************************************************************************************** const int maxnode = 4000 * 100 + 10; const int sigma_size = 26; // 字母表为全体小写字母的Trie struct Trie { int ch[maxnode][sigma_size]; int val[maxnode]; int sz; // 结点总数 void clear() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); } // 初始时只有一个根结点 int idx(char c) { return c - 'a'; } // 字符c的编号 // 插入字符串s,附加信息为v。注意v必须非0,因为0代表“本结点不是单词结点” void insert(const char *s, int v) { int u = 0, n = strlen(s); for(int i = 0; i < n; i++) { int c = idx(s[i]); if(!ch[u][c]) { // 结点不存在 memset(ch[sz], 0, sizeof(ch[sz])); val[sz] = 0; // 中间结点的附加信息为0 ch[u][c] = sz++; // 新建结点 } u = ch[u][c]; // 往下走 } val[u] = v; // 字符串的最后一个字符的附加信息为v } // 找字符串s的长度不超过len的前缀 bool find(const char *s, int len) { int u = 0; for(int i = 0; i < len; i++) { if(s[i] == '\0') break; int c = idx(s[i]); if(!ch[u][c]) break; u = ch[u][c]; if(val[u] != 0) return true; // 找到一个前缀 } return false; } }; //***************************************************************************************************** //以下为模板测试 Trie trie; const int maxl = 300000 + 10; // 文本串最大长度 const int maxw = 4000 + 10; // 单词最大个数 const int maxwl = 100 + 10; // 每个单词最大长度 char text[maxl], word[maxwl]; int main() { int n,m; scanf("%d",&n); trie.clear(); while(n--){ scanf("%s",word); trie.insert(word,1); } scanf("%d",&m); while(m--){ scanf("%s",text); int l=strlen(text); if(trie.find(text,l)) printf("\"%s\" in it\n",text); else printf("\"%s\" not in it\n",text); } return 0; }