码迷,mamicode.com
首页 > 其他好文 > 详细

HDU 2896

时间:2017-12-13 23:24:37      阅读:111      评论:0      收藏:0      [点我收藏+]

标签:mes   []   ace   使用   cstring   log   ons   for   markdown   

本题是一道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;
}

HDU 2896

标签:mes   []   ace   使用   cstring   log   ons   for   markdown   

原文地址:http://www.cnblogs.com/Mr-WolframsMgcBox/p/8034608.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!