HDU 1251 统计难题【字典树】

题意：中文题--跟着模板敲的--第一棵字典树[email protected][email protected]

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 typedef struct node
 7 {
 8     int cnt;
 9     struct node *next[26];
10 } Trie;
11 Trie *create()
12 {
13     Trie *p=(Trie *)(malloc)(sizeof(Trie));
14     p->cnt=1;
15     for(int i=0;i<26;i++) p->next[i]=NULL;
16     return p;
17 }
18
19 void insert(Trie *p,char *str)
20 {
21     int i=0;
22     while(str[i])
23     {
24         int idx=str[i]-‘a‘;
25         if(p->next[idx]) p->next[idx]->cnt++;
26         else p->next[idx]=create();
27         p=p->next[idx];
28         i++;
29     }
30 }
31 int search(Trie *p,char *str)
32 {
33     int len=strlen(str);
34     for(int i=0;i<len;i++)
35     {
36         int idx=str[i]-‘a‘;
37         p=p->next[idx];
38         if(p==NULL) return 0;
39     }
40     return p->cnt;
41 }
42
43 int main()
44 {
45     Trie *p=create();
46     char str[15];
47     while(gets(str)&&str[0])
48     insert(p,str);
49     while(scanf("%s",str)!=EOF)
50     printf("%d\n",search(p,str));
51 }