hdu 2896 ac自动机

#include <bits/stdc++.h>
#define eps 1e-14
#define pi acos(-1)
#define ll long long
#define RD T*(rand()*2-RAND_MAX)
#define Drand (long double)rand()/RAND_MAX
#define LINF 0x7f7f7f7f7f7f7f7f
#define INF 0x3f3f3f3f
using namespace std;
const int maxn=5e4+9000;
const long long mod=1e9+7;
ll MOD(ll a,ll m){return a>m?a%m+m:a;}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}

int trie[maxn][130],fail[maxn],cnt[maxn],bis[maxn];
int tot=0;
vector<int>ans;
char a[10005];


void insert(char s[],int vis)
{
    int len=strlen(s);
    int p=0;
    for(int i=0;i<len;i++){
        int c=(int)s[i];
        if(trie[p][c]==0){
            trie[p][c]=++tot;
        }
        p=trie[p][c];
    }
    cnt[p]=vis;
}

void get_fail()
{
    queue<int>q;
    for(int i=0;i<=128;i++){
        int x=trie[0][i];
        if(x!=0){
            fail[x]=0;
            q.push(x);
        }
    }
    while(!q.empty()){
        int now=q.front();
        q.pop();
        for(int i=0;i<=128;i++){
            int x=trie[now][i];
            if(x!=0){
                fail[x]=trie[ fail[now] ][i];
                q.push(x);
            }
            else{
                trie[now][i]=trie[ fail[now] ][i];
            }
        }
    }
}

void query(char s[])
{
    ans.clear();
    memset(bis,0,sizeof bis);
    int p=0;
    int len=strlen(s);
    for(int i=0;i<len;i++){
        int c=s[i];
        p=trie[p][c];
        for(int j=p;j!=0 && bis[j]==0;j=fail[j]){
            bis[j]=1;
            if(cnt[j]!=0){
                ans.push_back(cnt[j]);
            }
        }
    }
}



int main()
{
//    freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    int n,m,all=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",a);
        insert(a,i);
    }
    get_fail();
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",a);
        query(a);
        sort(ans.begin(),ans.end());
        int len=ans.size();
        if(len==0)continue;
        all++;
        printf("web %d:",i);
        for(int i=0;i<len;i++){
            printf(" %d",ans[i]);
        }printf("\n");
    }
    printf("total: %d\n",all);
    return 0;
}