病毒侵袭 HDU

病毒侵袭

 HDU - 2896 

题意：统计长串中出现了哪些短串

用vector存一下，再输出

#include <bits/stdc++.h>
using namespace std;
#define CLR(m,a) memset(m,a,sizeof(m))

const int maxnode=500*200+10;
const int sigma=128;

struct AC
{
    int ch[maxnode][sigma],last[maxnode],f[maxnode],val[maxnode];
    int sz;
    vector<int> ans;

    void init()
    {
        CLR(ch[0],0);
        val[0]=0;
        sz=1;
    }
    int idx(char c){return c;}

    void inser(char *s,int v)
    {
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                CLR(ch[sz],0);
                val[sz]=0;
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }

    void getfail()
    {
        queue<int> q;
        f[0]=0;
        for(int c=0;c<sigma;c++){
            int u=ch[0][c];
            if(u){
                q.push(u);
                f[u]=0;
                last[u]=0;
            }
        }
        while(!q.empty()){
            int r=q.front();
            q.pop();
            for(int c=0;c<sigma;c++){
                int u=ch[r][c];
                if(!u) continue;
                q.push(u);
                int 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 u)
    {
        if(u){
            ans.push_back(val[u]);
            print(last[u]);
        }
    }
    void fin(char *s)
    {
        ans.clear();
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            while(u&&!ch[u][c]) u=f[u];
            u=ch[u][c];
            if(val[u]) print(u);
            else if(last[u]) print(last[u]);
        }
    }
};
AC ac;
char s[210],t[10010];
int n,m;

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        ac.init();
        int cnt=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s);
            ac.inser(s,i);
        }
        ac.getfail();
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            scanf("%s",t);
            ac.fin(t);
            for(int j=0;j<ac.ans.size();j++)
            {
                sort(ac.ans.begin(),ac.ans.end());
                if(j==0){
                    printf("web %d: %d",i+1,ac.ans[j]);
                }
                else printf(" %d",ac.ans[j]);
                if(j==ac.ans.size()-1) cnt++,puts("");
            }
        }
        printf("total: %d
",cnt);
    }
}