HUST 1069 AntiVirus（AC自动机）

题目链接：http://acm.hust.edu.cn/problem.php?id=1609

题目很好，数据很水，直接解压，解压长度为给定病毒中最长的就OK了

但是实际是解压长度在0-50之间都能AC，也就是直接去掉压缩部分就可以了！

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <queue>
using namespace std;
#define maxn 1001000
#define MAX(a,b) (a>b?a:b)
#define MIN(a,b) (a<b?a:b)
char s[maxn],str[maxn*100];
int n,pos;
struct trie{
    trie *next[26],*fail;
    int num;
}po[maxn/4],re_root;
queue<trie *> q;
int insert_trie(trie *root,char *name){
    if(name[0]==0){
        root->num=1;
        return 0;
    }
    if(root->next[name[0]-'a']==NULL){
        memset(po+pos,0,sizeof(trie));
        root->next[name[0]-'a']=&po[pos++];
    }
    insert_trie(root->next[name[0]-'a'],name+1);
    return 0;
}
int parse_int(char *name,int &i,char &ch){
    int ans=0;
    i=0;
    while(name[i]>='0' && name[i]<='9'){
        ans=ans*10+name[i]-'0';
        i++;
    }
    ch=name[i];
    return ans;
}
int depress(char *s,char *str,int len){
    int i=0,j,k=0,t,l;
    char ch;
    while(s[i]){
        if(s[i]=='['){
            t=parse_int(s+i+1,l,ch);
            t=MIN(t,len);
            while(t--) str[k++]=ch;
            i+=l+3;
        }
        else str[k++]=s[i++];
    }
    str[k]=0;
    return 0;
}
int build_ac_auto(){
    q.push(&re_root);
    trie *p,*temp;
    re_root.fail=NULL;
    while(!q.empty()){
        p=q.front(),q.pop();
        for(int i=0;i<26;i++){
            if(p->next[i]){
                temp=p->fail;
                while(temp && temp->next[i]==NULL) temp=temp->fail;
                if(temp==NULL) p->next[i]->fail=&re_root;
                else p->next[i]->fail=temp->next[i];
                q.push(p->next[i]);
            }
        }
    }
    return 0;
}
int query(trie *root,char *name){
    int ans=0,i=0,j,k,t;
    trie *p=root,*temp;
    while(name[i]){
        t=name[i]-'a';
        while(p!=NULL && p->next[t]==NULL) p=p->fail;
        if(p==NULL) p=root;
        else p=p->next[t];
        temp=p;
        while(temp && temp->num!=-1){
            ans+=temp->num;
            temp->num=-1;
            temp=temp->fail;
        }
        i++;
        if(ans==n)
        break;
    }
    if(ans==0)
    printf("No %d\n",ans);
    else printf("Yes %d\n",ans);
    return 0;
}
int main(){
    int i,j,k,t,len;
    scanf("%d",&t);
    while(t--){
        pos=0;
        scanf("%d",&n);
        len=0;
        memset(&re_root,0,sizeof(trie));
        for(i=0;i<n;i++){
            scanf("%s",s);
            insert_trie(&re_root,s);
            k=strlen(s);
            len=MAX(k,len);
        }
        build_ac_auto();
        scanf("%s",s);
        if(len>=1) len=1;/*这里的len在0-50之间都能AC表示数据不带这么坑的*/
        depress(s,str,len);
        query(&re_root,str);
    }
    return 0;
}