Detect the Virus - ZOJ 3430（恶心的自动机）

题目大意：给你一些病毒的特征码，然后再给一些文本，判断每个文本有多少种病毒，不过给的字符串都是加密处理过的，给的每个字符串都有对应一个64以内的一个数（题目里面那个表就是），然后可以把这个64以内的这个数化成6位的二进制数，然后把这些二进制数每8位再化成一个字符，这就是原来的字符，比如 QA== ->编号0 1->二进制010000 000000->每8位变成一个字符 01000000 0000（后面这4个0就是那两个==，可以舍去）-> 64('@')。

 

分析：因为是化成8位的二进制，2^8 = 256，不过因为有‘\0’这样的字符，所以无法使用char来搞，开一个int数组是个不错的选择，当然也得注意内存开销，申请内存的方式会MLE，实验得出结果开6W最好.....真是让人感觉很恶心，错了19次！！

 

代码如下：

===============================================================================================================================

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;

const int MAXN = ;
const int MAXM = ;///2^8
const int oo = 1e9+;

const char cb64[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
int virus[MAXN], MumStr[MAXN];
char s[MAXN];
int password[MAXN*];

struct node
{
    int Fail, next[MAXM];
    int leaf, num;

    void InIt()
    {
        Fail = leaf = num = ;
        memset(next, , sizeof(next));
    }

}trie[];
int top;

int TurnStr(char s[], int p[])
{
    int i, j, k=;

    for(i=; s[i] && s[i] != '='; i++)
    {
        s[i] = strchr(cb64, s[i]) - cb64;

        for(j=; j>=; j--)
        {
            password[k+j] = s[i] % ;
            s[i] /= ;
        }

        k += ;
    }

    int len = ;

    for(i=j=; i<k; i++)
    {
        j = j* + password[i];

        if((i+) %  == )
        {
            p[len++] = j;
            j = ;
        }
    }

    return len;
}
void Insert(int root, int N)
{
    int p = root;

    for(int i=; i<N; i++)
    {
        int k = virus[i];

        if(trie[p].next[k] == )
        {
            trie[p].next[k] = ++top;
            trie[top].InIt();
        }
        p = trie[p].next[k];
    }

    trie[p].leaf += ;
}
void GetFail(int root)
{
    int p = root, temp;
    queue<int> Q;

    trie[root].Fail = -;

    for(int i=; i<MAXM; i++)if(trie[p].next[i])
    {
        int k = trie[p].next[i];

        trie[k].Fail = root;
        Q.push(k);
    }

    while(Q.size())
    {
        p = Q.front();
        Q.pop();

        for(int i=; i<MAXM; i++)if(trie[p].next[i])
        {
            int k = trie[p].next[i];
            temp = trie[p].Fail;

            while(temp != -)
            {
                if(trie[temp].next[i])
                {
                    trie[k].Fail = trie[temp].next[i];
                    break;
                }

                temp = trie[temp].Fail;
            }

            if(temp == -)
                trie[k].Fail = root;

            Q.push(k);
        }
    }
}
int Query(int root, int N, int num)
{
    int sum = ;
    int p = root, temp;

    for(int i=; i<N; i++)
    {
        int k = MumStr[i];

        while(!trie[p].next[k] && p!=root)
            p = trie[p].Fail;

        if(!trie[p].next[k])continue;

        temp = p = trie[p].next[k];

        while(temp != root && trie[temp].num != num)
        {
            if(trie[temp].leaf)
                sum += ;
            trie[temp].num = num;
            temp = trie[temp].Fail;
        }
    }

    return sum;
}

int main()
{
    int N, M;

    while(scanf("%d", &N) != EOF)
    {
        int root = ;

        trie[].InIt();
        top = ;

        for(int i=; i<=N; i++)
        {
            scanf("%s", s);
            int len = TurnStr(s, virus);

            Insert(root, len);
        }

        GetFail(root);
        scanf("%d", &M);

        for(int i=; i<=M; i++)
        {
            scanf("%s", s);
            int len = TurnStr(s, MumStr);

            printf("%d\n", Query(root, len, i));
        }
        printf("\n");
    }

    return ;
}