HDU 2896 病毒侵袭

本题是一道AC自动机的裸题，但是本题有许多天坑

MLE

1.使用指针写的请出门右转交 c++

WA

2.输出病毒的序号必须按照从小到大的顺序
3.输入的字符串中可能会有空格，要使用gets

PE

4.每一行的行末不能有多余空格
5.结尾要输出一个换行符

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <queue>
#define RST(a) memset((a),0,sizeof((a)))
using namespace std;
const int MAXN=100005;
int n,m,cnt;
char a[MAXN];
struct ACAM{
    int nxt[MAXN][128],lst[MAXN],end[MAXN],nume,fail[MAXN],vis[MAXN];
    void clear(){
        RST(nxt);RST(lst);RST(end),nume=0;RST(vis);
    }
    void ins(char s[],int index){
        int len=strlen(s);
        int p=0;
        for(int i=0;i<len;i++){
            int v=s[i];
            if(!nxt[p][v]) nxt[p][v]=++nume;
            p=nxt[p][v];
        }
        end[p]=index;
    }
    void getfail(){
        queue <int> q;
        for(int i=0;i<128;i++) if(nxt[0][i]) q.push(nxt[0][i]);
        int p=0;
        while(!q.empty()){
            p=q.front(),q.pop();
            for(int i=0;i<=127;i++){
                int v=nxt[p][i];
                if(v){
                    fail[v]=nxt[fail[p]][i],q.push(v);
                    lst[v]=end[fail[v]]?fail[v]:lst[fail[v]];
                }else nxt[p][i]=nxt[fail[p]][i];
            }
        }
    }
    void query(char s[],int index){
        int len=strlen(s);
        int ans[100];
        RST(ans);
        int p=0;
        for(int i=0;i<len;i++){
            p=nxt[p][s[i]];
            if(end[p]&&vis[p]!=index){
                ans[++ans[0]]=end[p];vis[p]=index;
            }
            for(int k=lst[p];k&&vis[k]!=index;k=lst[k]) {
                ans[++ans[0]]=end[k],vis[k]=index;
            }
        }
        if(ans[0]){
            cnt++;
            printf("web %d: ",index);
            sort(ans+1,ans+1+ans[0]);
            for(int i=1;i<ans[0];i++) printf("%d ",ans[i]);
            printf("%d\n",ans[ans[0]]);
        }
    }
}AC;
int main(){
    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    getchar();
    for(int i=1;i<=n;i++){
        gets(a);
        AC.ins(a,i);
    }
    AC.getfail();
    scanf("%d",&m);
    getchar();
    for(int i=1;i<=m;i++){
        gets(a);
        AC.query(a,i);
    }
    printf("total: %d\n",cnt);
    fclose(stdin);
    return 0;
}

转载于:https://www.cnblogs.com/Mr-WolframsMgcBox/p/8034608.html