HDU 2846 Repository（字典树）

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 
 5 using namespace std;
 6 
 7 int ch[500005][30];
 8 int flag[500005];
 9 int val[500005];
10 int cnt = 1;
11 
12 inline int ids(char c){
13     return c - ‘a‘;
14 }
15 
16 inline void build(char *s, int k){
17     int u = 0;
18     int len = strlen(s);
19     for(int i = 0; i < len; i++){
20         int id = ids(s[i]);
21         if(!ch[u][id]) ch[u][id] = cnt++;
22         u = ch[u][id];
23         if(flag[u] != k){
24             val[u]++;
25             flag[u] = k;
26         }
27     }
28 }
29 
30 inline int query(char *s){
31     int len = strlen(s);
32     int u = 0;
33     for(int i = 0; i < len; i++){
34         int id = ids(s[i]);
35         if(!ch[u][id]) return 0;
36         u = ch[u][id];
37     }
38     return val[u];
39 }
40 
41 int main(){
42     int p, q;
43     char s[25];
44     scanf("%d", &p);
45     memset(flag, -1, sizeof(flag));
46     for(int i = 1; i <= p; i++){
47         scanf("%s", s);
48         int len = strlen(s);
49         for(int j = 0; j < len; j++)
50             build(s + j, i);
51     }
52     scanf("%d", &q);
53     for(int i = 0; i < q; i++){
54         scanf("%s", s);
55         printf("%d\n", query(s));
56     }
57     return 0;
58 }