BZOJ 4327: JSOI2012 玄武密码 后缀自动机

Description

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

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

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

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

Input

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

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

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

 

Output

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

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

题解： 

后缀自动机模板题. 

注意只有 4 种字符，简单预处理一下每种字符映射的数字即可.(开一个 map 也行）

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 18000001
using namespace std;
char str[maxn];  
int idx(char i)
{
    if(i=='S') return 0;
    if(i=='N') return 1; 
    if(i=='W') return 2; 
    if(i=='E') return 3; 
}
namespace SAM
{
    int ch[maxn][4],f[maxn],dis[maxn]; 
    int tot,last;
    void init()  
    { 
        tot=last=1;  
    }
    void ins(int c)
    {
        int np=++tot,p=last;   
        last=np,dis[np]=dis[p]+1; 
        while(p&&!ch[p][c]) ch[p][c]=np,p=f[p]; 
        if(!p) f[np]=1;                      
        else     
        {
            int q=ch[p][c];
            if(dis[q]==dis[p]+1) f[np]=q;
            else 
            {     
                int nq=++tot;
                dis[nq]=dis[p]+1;     
                memcpy(ch[nq],ch[q],sizeof(ch[q])); 
                f[nq]=f[q],f[q]=f[np]=nq;                  
                while(p&&ch[p][c]==q) ch[p][c]=nq,p=f[p];      
            }
        }
    } 
}; 
int main()
{
    // setIO("input"); 
    int n,m;
    scanf("%d%d%s",&n,&m,str+1);
    SAM::init();  
    for(int i=1;i<=n;++i) SAM::ins(idx(str[i]));     
    while(m--)
    { 
        scanf("%s",str+1);
        int k=strlen(str+1),p=1,cur=0;
        for(int i=1;i<=k;++i)
        {
            int c=idx(str[i]); 
            if(!SAM::ch[p][c]) 
                break; 
            else 
                ++cur,p=SAM::ch[p][c]; 
        }
        printf("%d\n",cur); 
    }
    return 0; 
}