hdu----（2848）Repository（trie树变形）

Repository

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 2538    Accepted Submission(s): 990

Problem Description

When
you go shopping, you can search in repository for avalible merchandises
by the computers and internet. First you give the search system a name
about something, then the system responds with the results. Now you are
given a lot merchandise names in repository and some queries, and
required to simulate the process.

 

Input

There
is only one case. First there is an integer P
(1<=P<=10000)representing the number of the merchanidse names in
the repository. The next P lines each contain a string (it's length
isn't beyond 20,and all the letters are lowercase).Then there is an
integer Q(1<=Q<=100000) representing the number of the queries.
The next Q lines each contains a string(the same limitation as foregoing
descriptions) as the searching condition.

 

Output

For each query, you just output the number of the merchandises, whose names contain the search string as their substrings.

 

Sample Input

20
ad
ae
af
ag
ah
ai
aj
ak
al
ads
add
ade
adf
adg
adh
adi
adj
adk
adl
aes
5
b
a
d
ad
s

 

Sample Output

0
20
11
11
2

 

Source

2009 Multi-University Training Contest 4 - Host by HDU

 

题意： 给出一些字符，然后寻问一个字符，在给出字符里能找到子串的个数..

 

代码：

 //#define LOCAL
 #include<cstdio>
 #include<cstring>
 typedef struct node
 {
    struct node *child[];
    int cnt;  //作为统计
    int id;
 }Trie;

 void Insert(char *s,Trie *root,int id)
 {
     int pos,i;
     Trie *cur=root,*curnew;
     for(;*s!='\0';s++)
     {
         pos=*s-'a';
       if(cur->child[pos]==NULL)
       {
            curnew = new Trie;
            for( i=; i<;i++)
             curnew->child[i]=NULL;
             curnew->cnt=;
             curnew->id=;
             cur->child[pos]=curnew;
       }
       cur=cur->child[pos];
       if(cur->id!=id) //避免同一个单词重复计算子串
          {
              cur->cnt++;
              cur->id=id;
          }
     }
 }

 int query(char *s, Trie *root)
 {
     int pos;
     Trie *cur=root;
     while(*s!='\0')
     {
         pos=*s-'a';
         if(cur->child[pos]==NULL)
             return ;
         cur=cur->child[pos];
         s++;
     }
     return cur->cnt;
 }
 void del(Trie *root)
 {
     Trie *cur=root;
     for(int i=;i<;i++)
     {
         if(cur->child[i]!=NULL)
            del(cur->child[i]);
     }
     delete cur;
   return ;
 }
 char str[];
 int main()
 {
 #ifdef LOCAL
   freopen("test.in","r",stdin);
 #endif
   int n,m,i;
   scanf("%d",&n);
   Trie *root=new Trie;
   for( i=;i<;i++)
     root->child[i]=NULL;
     root->cnt=;
   while(n--)
   {
       scanf("%s",str);
       for(i= ; str[i]!='\0' ;i++)
         Insert(str+i,root,n+);
   }
   scanf("%d",&m);
   while(m--)
   {
       scanf("%s",str);
       printf("%d\n",query(str,root));
   }
   del(root);
   return ;
 }