hdu 2846 统计给定字符串是多少字符串的子串

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2846

给定一些字符串，约10000个，给定q的询问，每次询问有一个字符串，判断这个字符串是多少字符串的子串。

字典树处理，把给定字符串的每个后缀都插入字典树，要注意不要把相同的子串重复计数了，因此设置一个id标记，判断之前一个计数是不是当前字符串计数的。

另外字典树的数组要设置的大一些，，，不然会TLE。

#include<bitset>  //G++ 109 ms
#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#define inf 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;

inline int in()
{
    int res=0;char c;
    while((c=getchar())<'0' || c>'9');
    while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
    return res;
}
const int N=1000010;

struct trie
{
    int next[27];
    bool flag;
    int num,id;
    void init()
    {
        flag=0;
        num=0;
        id=-inf;
        mem(next,-1);
    }
}trie[N];

char word[22];
int tot;

void insert(int i,int id,int n)
{

    int now=0,tmp,next;
    for(;i<n;i++)
    {
        tmp=word[i]-'a';
        next=trie[now].next[tmp];
        if(next==-1)
        {
            next = ++tot;
            trie[next].init();
            trie[now].next[tmp]=next;
        }
        if(trie[now].id!=id)
        {
            trie[now].id=id;
            trie[now].num++;
        }
        now=next;
    }
    if(trie[now].id!=id)
    {
        trie[now].id=id;
        trie[now].num++;
    }
    trie[now].flag=1;

}

int query()
{
    int n=strlen(word);
    int now=0,tmp,next;
    for(int i=0;i<n;i++)
    {
        tmp=word[i]-'a';
        next=trie[now].next[tmp];
        if(next==-1) return 0;
        now=next;
    }
    return trie[now].num;
}

int main()
{
    int T=in();
    tot=0;

    trie[0].init();

    while(T--)
    {
        gets(word);
        int n=strlen(word);
        for(int i=0;i<n;i++)
            insert(i,T,n);
    }
    int q=in();
    while(q--)
    {
        gets(word);
        printf("%d\n",query());
    }
    return 0;
}