hdu2222

这题wa了好久，然后再discuss里看了一下，查询的字符串居然可以重复出现，也就是说

1

2

a

a

a

应该是2不是1。。。。 感觉再也不会爱了==

#include<iostream>
#include<cstdio>
#include<string.h>
#include<queue>
using namespace std;
const int maxn=10001;
char tree[51],p[maxn*100];
int cnt,val[maxn*50],f[maxn*50],last[maxn*50];
bool vis[maxn*50];
int n,top;
int ch[maxn*50][26];

void init()
{
    top=1;
    cnt=0;
    memset(ch[0],0,sizeof(ch[0]));
    memset(val,0,sizeof(val));
    memset(vis,0,sizeof(vis));
}

void insert(char *s,int op)
{
    int n=strlen(s);
    int u=0,c,i;
    for(i=0;i<n;i++)
    {
        c=s[i]-'a';
        if(ch[u][c]==0)
        {
            memset(ch[top],0,sizeof(ch[top]));
            val[top]=0;
            ch[u][c]=top++;
        }
        u=ch[u][c];
    }
    val[u]++;
    //printf("op=%d val=%d u=%d\n",op,val[u],u);
}

void getfail()
{
    queue<int> q;
    int u,c;
    for(c=0;c<26;c++)
    {
        u=ch[0][c];
        if(u){f[u]=0;q.push(u);last[u]=0;}
    }
    int r,v;
    while(!q.empty())
    {
        r=q.front();q.pop();
        for(c=0;c<26;c++)
        {
            u=ch[r][c];
            if(!u) continue;
            q.push(u);
            v=f[r];
            while(v&&!ch[v][c]) v=f[v];
            f[u]=ch[v][c];
            last[u]=val[f[u]]?f[u]:last[f[u]];
        }
    }
}

void print(int j)
{
    if(j)
    {
        if(!vis[j])
        cnt+=val[j];
        vis[j]=1;
        print(last[j]);
    }
}
void find(char *T)
{
    int n=strlen(T),i,j=0,c;
    for(i=0;i<n;i++)
    {
        c=T[i]-'a';
        while(j&&!ch[j][c]) j=f[j];
        j=ch[j][c];
        if(val[j])  print(j);
        else if(last[j]) print(last[j]);
    }
}
int main()
{
    //freopen("test.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int i;
        init();
        for(i=0;i<n;i++)
        {
            scanf("%s",tree);
            insert(tree,i+1);
        }
        getfail();
        scanf("%s",p);
        find(p);
        printf("%d\n",cnt);
    }
    return 0;
}