BZOJ-4327：JSOI2012 玄武密码（AC自动机模板题）

在美丽的玄武湖畔，鸡鸣寺边，鸡笼山前，有一块富饶而秀美的土地，人们唤作进香河。相传一日，一缕紫气从天而至，只一瞬间便消失在了进香河中。老人们说，这是玄武神灵将天书藏匿在此。

很多年后，人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是，这份带有玄武密码的文字，与玄武湖南岸台城的结构有微妙的关联。于是，漫长的破译工作开始了。

经过分析，我们可以用东南西北四个方向来描述台城城砖的摆放，不妨用一个长度为N的序列来描述，序列中的元素分别是‘E’，‘S’，‘W’，‘N’，代表了东南西北四向，我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象，分别是东之青龙，西之白虎，南之朱雀，北之玄武，对东南西北四向相对应。

现在，考古工作者遇到了一个难题。对于每一段文字，其前缀在母串上的最大匹配长度是多少呢？

Input

第一行有两个整数，N和M，分别表示母串的长度和文字段的个数。

第二行是一个长度为N的字符串，所有字符都满足是E,S,W和N中的一个。

之后M行，每行有一个字符串，描述了一段带有玄武密码的文字。依然满足，所有字符都满足是E,S,W和N中的一个。

Output

输出有M行，对应M段文字。

每一行输出一个数，表示这一段文字的前缀与母串的最大匹配串长度。

Sample Input7 3 SNNSSNS NNSS NNN WSEE

Sample Output4 2 0Hint

对于100%的数据，N<=10^7，M<=10^5，每一段文字的长度<=100。

应上传者要求，此题不公开，如有异议，请提出.

思路：AC自动机，一开始以为不能一直沿着Fail指针上传，会TLE，需要优化。但其实每个点最多标记一次，所以整体复杂度是O(Lc+Ls)的。

所以暴力搞就行了。

#include<bits/stdc++.h>

using namespace std;

const int maxn=;

char c[maxn],s[maxn];

int ch[maxn][],fa[maxn],pos[maxn],vis[maxn];

int q[maxn],dep[maxn],fail[maxn],cnt,head,tail;

int get(char chr){ if(chr=='E') return ; if(chr=='S') return ; if(chr=='W') return ; return ;}

void insert(int opt)

{

    int Now=,d=;

    for(int i=;c[i];i++){

        d++;

        if(!ch[Now][get(c[i])]) {

            ch[Now][get(c[i])]=++cnt;

            fa[cnt]=Now; dep[cnt]=d;

        }

        Now=ch[Now][get(c[i])];

    }   pos[opt]=Now;

}

void failbuild()

{

    for(int i=;i<;i++) if(ch[][i]) q[++head]=ch[][i];

    while(tail<head){

        int Now=q[++tail];

        for(int i=;i<;i++){

            if(ch[Now][i]) {

                fail[ch[Now][i]]=ch[fail[Now]][i];

                q[++head]=ch[Now][i];

            }

            else ch[Now][i]=ch[fail[Now]][i];

        }

    }

}

int main()

{

    int L,N,i,j;

    scanf("%d%d%s",&L,&N,s+);

    for(i=;i<=N;i++){

        scanf("%s",c+);

        insert(i);

    }

    failbuild();

    int Now=;

    for(i=;i<=L;i++){

        Now=ch[Now][get(s[i])];

        int tmp=Now;

        while(tmp&&!vis[tmp]){

            vis[tmp]=; tmp=fail[tmp];

        }

    }

    for(i=;i<=N;i++){

        Now=pos[i];

        while(!vis[Now]&&Now) Now=fa[Now];

        printf("%d\n",dep[Now]);

    }

    return ;

}

巴特西

BZOJ-4327：JSOI2012 玄武密码（AC自动机模板题）

最新文章

热门文章