zoj 3430 Detect the Virus

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4114

题目大意及思路：ac自动机，不过要先解码，需要注意的是这里有256种字符，需要用无符号字符型，而且0也不能作为字符串的结束标志了。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 550
int max(int a,int b)
{
	return a>b?a:b;
}
int min(int a,int b)
{
	return a<b?a:b;
}
int q[70*Max],cnt,rec,len;
unsigned char s[2200*8];
unsigned char s2[2200*8];
unsigned char mp[100][8];
struct node
{
    int cnt,fail,flag;
    int next[256];
    void init()
    {
        flag=cnt=fail=0;
        memset(next,0,sizeof(next));
    }
}tri[70*Max];
void decode(unsigned char *s,unsigned char *str)
{
    int i,num,tmp,tmplen;
    str[0]=0,num=0,len=0,tmplen=0;
    for(i=0;s[i];i++)
    {
        if(s[i]=='=')
        {
            num++;
            continue;
        }
        if(s[i]>='A'&&s[i]<='Z')
            tmp=s[i]-'A';
        if(s[i]>='a'&&s[i]<='z')
            tmp=s[i]-'a'+26;
        if(s[i]>='0'&&s[i]<='9')
            tmp=s[i]-'0'+52;
        if(s[i]=='+')
            tmp=62;
        if(s[i]=='/')
            tmp=63;
        for(int j=5;j>=0;j--)
        {
            str[tmplen++]=((tmp&(1<<j))>0);
        }
    }
    if(num==2)
        tmplen-=4;
    else if(num==1)
        tmplen-=2;
    len=0;
     for(i=0;i<tmplen;i+=8)
        {
            tmp=0;
            for(int j=0;j<8;j++)
            {
                tmp+=str[i+j]*(1<<(7-j));
            }
            s[len++]=tmp;

        }
}
void insert(unsigned char *s)
{
    int i,x,p;
    p=0;
    for(i=0;i<len;i++)
    {
        x=s[i];
        if(!tri[p].next[x])
        {
            tri[++cnt].init();
            tri[p].next[x]=cnt;
        }
        p=tri[p].next[x];
    }
    tri[p].cnt++;
}
void bfs()
{
    int i,p,suf,head=0,tail=0;
    p=0;
    for(i=0;i<256;i++)
    {
        if(tri[0].next[i])
        {
            q[tail++]=tri[0].next[i];
            tri[q[tail-1]].fail=0;
        }
    }
    while(head<tail)
    {
        p=q[head++];suf=tri[p].fail;
        for(i=0;i<256;i++)
        {
            if(tri[p].next[i])
            {
                q[tail++]=tri[p].next[i];
                tri[q[tail-1]].fail=tri[suf].next[i];
            }
            else
                tri[p].next[i]=tri[suf].next[i];
        }
    }
}
void query(unsigned char *s,int num)
{
    int i,x,p,tmp;
    p=0;
    rec=0;
    for(i=0;i<len;i++)
    {
        x=s[i];
        p=tri[p].next[x];
        tmp=p;
        while(tmp&&tri[tmp].flag!=num)
        {
            rec+=tri[tmp].cnt;
            tri[tmp].flag=num;
            tmp=tri[tmp].fail;
        }
    }
}
int main()
{
    int n,m,i;
    unsigned char str[80*8];
    unsigned char str2[80*8];
    while(scanf("%d",&n)!=EOF)
    {
        cnt=0;
        tri[0].init();

        for(i=1;i<=n;i++)
        {
            scanf("%s",str);
            decode(str,str2);
            insert(str);
        }
        bfs();
        scanf("%d",&m);
        for(i=1;i<=m;i++)
        {
            scanf("%s",s);
            decode(s,s2);
            query(s,i);
            printf("%d\n",rec);
        }
        puts("");
    }
}