HihoCode 1014 （字典树）

#1014 : Trie树

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

省略

输入

输入的第一行为一个正整数n，表示词典的大小，其后n行，每一行一个单词（不保证是英文单词，也有可能是火星文单词哦），单词由不超过10个的小写英文字母组成，可能存在相同的单词，此时应将其视作不同的单词。接下来的一行为一个正整数m，表示小Hi询问的次数，其后m行，每一行一个字符串，该字符串由不超过10个的小写英文字母组成，表示小Hi的一个询问。

在20%的数据中n, m<=10，词典的字母表大小<=2.

在60%的数据中n, m<=1000，词典的字母表大小<=5.

在100%的数据中n, m<=100000，词典的字母表大小<=26.

本题按通过的数据量排名哦～

输出

对于小Hi的每一个询问，输出一个整数Ans,表示词典中以小Hi给出的字符串为前缀的单词的个数。

样例输入

5
babaab
babbbaaaa
abba
aaaaabaa
babaababb
5
babb
baabaaa
bab
bb
bbabbaab

样例输出

1
0
3
0
0

代码：

 #include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <queue>
#define mem(p,k) memset(p,k,sizeof(p));
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define inf 0x6fffffff
#define LL long long
using namespace std;
const int mod=10007;
typedef struct tir{
    int num,cur;
    struct tir* next[26];
}tir;
tir *root;
char s[11000];
int dp[1100][1100],a[1100];

void init(tir* &p){
    p=(tir*)malloc(sizeof(tir));
    mem(p->next,0);
    p->cur=p->num=0;
}
void ins(char *s){
    int len=strlen(s),i=0;
    tir *p=root,*q;
    while(i<len){
        if(s[i]<'a'||s[i]>'z')break;
        if(!p->next[s[i]-'a']){
            tir* q;
            init(q);
            p->next[s[i]-'a']=q;
        }
        q=p;
        p=p->next[s[i]-'a'];
        p->num++;
        i++;
    }
}
int fin(char *s){
    tir *p=root;
    int i=0,len=strlen(s);
    while(i<len){
        if(!p->next[s[i]-'a'])return 0;
        p=p->next[s[i]-'a'];
        i++;
    }
    return p->num;
}
int main()
{
	int n,m;
	cin>>n;
	init(root);
	for(int i=0;i<n;i++){
        scanf("%s",s);
        ins(s);
	}
	cin>>m;
	for(int i=0;i<m;i++){
        scanf("%s",s);
        cout<<fin(s)<<endl;
	}

    return 0;
}