hdu-1251 统计难题

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 char a[100];
 5 int tree[400100][30];
 6 int color[400100];
 7 int k=1;
 8 
 9 void insert()
10 {
11     int p=0;
12     int len=strlen(a);
13     for(int i=0;i<len;i++){
14         int c=a[i]-‘a‘;
15         if(!tree[p][c]) {
16             tree[p][c]=k;
17             k++;
18         }
19         p=tree[p][c];
20         color[p]++;
21     }
22 }
23 
24 int query()
25 {
26     int p=0;
27     int len=strlen(a);
28     for(int i=0;i<len;i++){
29         int c=a[i]-‘a‘;
30         if(!tree[p][c]) return 0;
31         p=tree[p][c];
32     }
33     return color[p];
34 }
35 
36 int main()
37 {
38     while(gets(a)){
39         if(strlen(a)==0) break;
40         insert();
41     }
42     while(gets(a)){
43         if(strlen(a)==0) break;
44         printf("%d\n",query());
45     }
46     return 0;
47 }