hdu 2846 Repository (字典树)

Repository
Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/65536 K (Java/Others)
Total Submission(s): 7233    Accepted Submission(s): 2278

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

 

C/C++:

#include <map>
#include <queue>
#include <cmath>
#include <vector>
#include <string>
#include <cstdio>
#include <cstring>
#include <climits>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
const int my_max = 1e6 + 10;

struct node
{
    int id, cnt, next[26];
}my_node[my_max];
int flag = 0;

void my_init(int x)
{
    for (int i = 0; i < 26; ++ i)
        my_node[x].next[i] = -1;
}

void my_insert(char *s, int len, int x)
{
    int now = 0;
    for (int i = 0; i < len; ++ i)
    {
        if (my_node[now].next[s[i] - 'a'] == -1)
        {
            my_node[now].next[s[i] - 'a'] = ++ flag;
            now = flag;
            my_init(flag);
        }
        else now = my_node[now].next[s[i] - 'a'];
        if (my_node[now].id != x) my_node[now].cnt ++;
        my_node[now].id = x;
    }
}

int my_find(char *s)
{
    int now = 0, len = strlen(s);
    for (int i = 0; i < len; ++ i)
    {
        if (my_node[now].next[s[i] - 'a'] == -1) return 0;
        now = my_node[now].next[s[i] - 'a'];
    }
    return my_node[now].cnt;
}

int main()
{
    int n, len;
    scanf("%d", &n);
    char s[25];
    my_init(0);
    while (n --)
    {
        scanf("%s", s);
        len = strlen(s);
        for (int i = 0; i < len; ++ i)
            my_insert(s + i, len - i, n + 1);
    }
    scanf("%d", &n);
    while (n --)
    {
        scanf("%s", s);
        printf("%d\n", my_find(s));
    }
    return 0;
}