思路：
完全看题目中的介绍就行了。还有里面的input写道：不保证是英文单词，也有可能是火星文单词哦。比赛结束后的提交是不用考虑26个字母之外的，都会AC，如果考虑128种可能的话，爆了内存。步骤就是，在插单词的同时记录该结点之后的单词数，查词就查最后一个字母所在结点上的单词数。
1 #include <iostream> 2 #include <cstring> 3 #include <vector> 4 #include <stdio.h> 5 using namespace std; 6 const int N=26; 7 char dict[15]; 8 int n, m; 9 struct node 10 { 11 int num; //以本节点开头的单词个数 12 node *child[N]; //孩子 13 }tree_gen; 14 15 int check_dict(char *p) 16 { 17 node *node_p=&tree_gen; //指向树根 18 int ans=0; 19 while(*p!=' ') 20 { 21 if(node_p->child[*p-'a']!=0) 22 { 23 node_p=node_p->child[*p-'a']; 24 ans=node_p->num; 25 } 26 else 27 return 0; 28 p++; 29 } 30 return ans; 31 } 32 33 int insert_tree(char *p) 34 { 35 node *node_p=&tree_gen; //指向树根 36 node_p->num++; //访问这个节点 37 while(*p!=' ') 38 { 39 if( node_p->child[*p-'a']==0 ) //还没有这叉，就要建 40 { 41 node *new_node=new(node); //创建新节点 42 for(int i=0; i<N; i++) 43 new_node->child[i]=0; 44 new_node->num=0; 45 node_p->child[*p-'a']=new_node; //连接工作 46 node_p=new_node; 47 } 48 else //已有这叉，继续往下 49 node_p=node_p->child[*p-'a']; 50 node_p->num++; //访问这个节点 51 p++; 52 } 53 return 0; 54 } 55 56 int main() 57 { 58 //freopen("e:input.txt", "r", stdin); 59 tree_gen.num=0; //树根的初始化 60 for(int i=0; i<N; i++) tree_gen.child[i]=0; 61 62 cin>>n;getchar(); 63 for(int i=0; i<n; i++) //接受字典,顺便插入，顺便记录num 64 { 65 gets(dict); 66 insert_tree(dict); 67 } 68 cin>>m;getchar(); 69 for(int i=0; i<m; i++) //查询以xxx开头的单词个数 70 { 71 gets(dict); 72 cout<<check_dict(dict)<<endl; 73 } 74 return 0; 75 }
还有一种不是以链表实现的树，而是用数组来做的。小Ho做的这代码可以过一些小一点的数据。
1 #include <cstdio> 2 #include <cstring> 3 4 const int MAXL = 10 + 2; 5 const int MAXN = 1000 + 10; 6 const int MAXM = 100000 + 10; 7 8 int next[MAXN * MAXL][26], count[MAXN * MAXL]; 9 10 int main() { 11 int n; 12 while (scanf("%d", &n) != EOF) { 13 int top = 1; memset(next, 0, sizeof(next)); 14 char str[MAXL]; 15 for (int i = 0; i < n; i++) { 16 scanf("%s", str); 17 int p = 1; 18 for (int j = 0; str[j] != ' '; j ++) { 19 if (next[p][str[j] - 'a'] == 0) next[p][str[j] - 'a'] = ++top; 20 p = next[p][str[j] - 'a']; 21 count[p] ++; 22 } 23 } 24 int m; 25 scanf("%d", &m); 26 while (m --) { 27 scanf("%s", str); 28 int p = 1; 29 for (int j = 0; str[j] != ' '; j++) { 30 p = next[p][str[j] - 'a']; 31 } 32 printf("%d ", count[p]); 33 } 34 } 35 }