HDU 3065 病毒侵袭持续中

毫无槽点，直水即可。

#include <algorithm>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>

#include <map>
#include <set>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <memory>
#include <complex>
#include <numeric>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <time.h>
#include <ctype.h>
#include <locale.h>

using namespace std;

#pragma pack(4)

const double  eps = 1e-8;
const double   pi = acos(-1.0);
const int     inf = 0x7f7f7f7f;
const __int64 INF = 0x7f7f7f7f7f7f7f7fll;

#define loop(a,n)                            \
    for(int i=0;n>i;i++)                     \
        cout<<a[i]<<(i!=n-1?' ':'\n')
#define loop2(a,n,m)                         \
    for(int i=0;n>i;i++)                     \
        for(int j=0;m>j;j++)                 \
            cout<<a[i][j]<<(j!=m-1?' ':'\n')

#define   at(a,i) ((a)&(1<<(i)))
#define   nt(a,i) ((a)^(1<<(i)))
#define set1(a,i) ((a)|(1<<(i)))
#define set0(a,i) ((a)&(~(1<<(i))))

#define gret(a,b) (((a)-(b))>eps)
#define less(a,b) (((b)-(a))>eps)
#define greq(a,b) (((a)-(b))>-eps)
#define leeq(a,b) (((b)-(a))>-eps)
#define equl(a,b) (fabs((a)-(b))<eps)

#define lmax(a,b) ((a)>(b)?(a):(b))
#define lmin(a,b) ((a)<(b)?(a):(b))
#define fmax(a,b) (gret(a,b)?(a):(b))
#define fmin(a,b) (less(a,b)?(a):(b))

const int MAXV = 2000002;

struct node
{
    int cnt,fail,next[26];
}trie[MAXV];

int _index,Q[MAXV],cnt[MAXV];

void clear(void)
{
    _index=0;
}
int malloc(void)
{
    trie[_index].cnt=0;
    trie[_index].fail=-1;
    for(int i=0;26>i;i++)
    {
        trie[_index].next[i]=-1;
    }
    return _index++;
}
int fail(int root,int c)
{
    while(trie[root].next[c]==-1)
    {
        if(trie[root].fail==-1) break;
        else root=trie[root].fail;
    }
    return root;
}
void insert(int root,char p[],int v)
{
    for(int i=0,l=strlen(p);l>i;i++)
    {
        if(trie[root].next[p[i]-'A']==-1) trie[root].next[p[i]-'A']=malloc();
        root=trie[root].next[p[i]-'A'];
    }
    trie[root].cnt=v;
}
void build(int root)
{
    Q[0]=root;
    for(int f=0,r=1;f!=r;)
    {
        int u=Q[f++];
        for(int i=0;26>i;i++)
        {
            if(trie[u].next[i]!=-1)
            {
                trie[Q[r++]=trie[u].next[i]].fail=root;
                if(u!=root)
                {
                    int v=fail(trie[u].fail,i);
                    if(trie[v].next[i]!=-1) trie[trie[u].next[i]].fail=trie[v].next[i];
                }
            }
        }
    }
}
void query(int root,char t[])
{
    for(int i=0,l=strlen(t);l>i;i++)
    {
        if('A'<=t[i]&&t[i]<='Z')
        {
            root=fail(root,t[i]-'A');
            if(trie[root].next[t[i]-'A']!=-1)
            {
                root=trie[root].next[t[i]-'A'];
                for(int u=root;u!=-1;u=trie[u].fail)
                {
                    cnt[trie[u].cnt]++;
                }
            }
        }
        else root=0;
    }
}

int n;
char a[1002][52],b[MAXV];

int main()
{
    #ifndef ONLINE_JUDGE
    freopen("3065.txt","r",stdin);
    #endif

    while(scanf("%d\n",&n)!=EOF)
    {
        clear();
        int root=malloc();
        memset(cnt,0,sizeof(cnt));

        for(int i=1;n>=i;i++)
        {
            scanf("%s",a[i]);
            insert(root,a[i],i);
        }
        build(root);
        scanf("%s",b);
        query(root,b);

        for(int i=1;n>=i;i++)
        {
            if(cnt[i]) printf("%s: %d\n",a[i],cnt[i]);
        }
    }
    return 0;
}