hdu 1251 (Trie水题）

统计难题

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 36295    Accepted Submission(s): 13515

Problem Description

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

 

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

 

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

 

Sample Input

banana
band
bee
absolute
acm

ba
b
band
abc

 

Sample Output

2
3
1
0

 

Author

Ignatius.L

 

Recommend

Ignatius.L   |   We have carefully selected several similar problems for you:  1075 1247 1671 1298 1800

---

这题就是Trie水题，纯练手用的。但是题目比较坑没给你单词书目。目测十万可过。下面是套模板的Trie:

 #include<cstdio>
 #include<iostream>
 #include<cstring>
 #define clr(x) memset(x,0,sizeof(x))
 #define clrmin(x) memset(x,-1,sizeof(x))
 using namespace std;
 struct node
 {
     int lt,rt;
     int val;
     int num;
     char *c;
 };
 struct Trie
 {
     int head,len;
     node tr[];
     Trie () { clr(tr); head=; len=;}
     void init()
     {
         clr(tr);
         head=;
         len=;
         return ;
     }
     int newnode(char *c)
     {
         if(!head)
         {
             head=len;
         }
         tr[len].c=c;
         return len++;
     }
     void push(int fa,int now,char *str)
     {
         int p;
         if(!now)
         {
             now=newnode(str);
             tr[fa].lt=now;
             tr[now].num++;
             if(str[]=='\0')
             {
                 tr[now].val++;
             }
             else
                 push(now,,str+);
             return ;
         }
         while(now && *tr[now].c!=*str)
         {
             p=now;
             now=tr[now].rt;
         }
         if(!now)
         {
             now=newnode(str);
             tr[p].rt=now;
             tr[now].num++;
         }
         else
             tr[now].num++;
         if(str[]=='\0')
         {
             tr[now].val++;
         }
         else
             push(now,tr[now].lt,str+);
         return;
     }
     int found(int now,char *str)
     {
         while(now && *tr[now].c!=*str)
         {
             now=tr[now].rt;
         }
         if(!now)
             return ;
         else
             if(strlen(str)==)
                 return tr[now].num;
             else
                 return found(tr[now].lt,str+);
     }
 }tree;
 char s[][],str[];
 int main()
 {
     int n=;
     tree.init();
     while(fgets(str,,stdin) && strcmp(str,"\n")!=)
     {
         str[strlen(str)-]='\0';
         strcpy(s[n++],str);
         tree.push(,tree.head,s[n-]);
     }
     while(fgets(str,,stdin)!=NULL)
     {
         str[strlen(str)-]='\0';
         printf("%d\n",tree.found(tree.head,str));
     }
     return ;
 }