BZOJ4327:[JSOI2012]玄武密码(SAM)

Description

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

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

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

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

Input

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

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

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

Output

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

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

Sample Input

7 3
SNNSSNS
NNSS
NNN
WSEE

Sample Output

4
2
0

HINT

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

Solution

$SAM$板子……

Code

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cstdio>
 4 #define N (20000007)
 5 using namespace std;
 6 
 7 int n,m,v[109];
 8 char s[N];
 9 
10 struct SAM
11 {
12     int p,q,np,nq,last,cnt;
13     int fa[N],son[N][4],step[N];
14     SAM() {last=cnt=1;}
15     
16     void Insert(int x)
17     {
18         p=last; np=last=++cnt; step[np]=step[p]+1;
19         while (p && !son[p][x]) son[p][x]=np, p=fa[p];
20         if (!p) fa[np]=1;
21         else
22         {
23             q=son[p][x];
24             if (step[q]==step[p]+1) fa[np]=q;
25             else
26             {
27                 nq=++cnt; step[nq]=step[p]+1;
28                 memcpy(son[nq],son[q],sizeof(son[q]));
29                 fa[nq]=fa[q]; fa[q]=fa[np]=nq;
30                 while (son[p][x]==q) son[p][x]=nq, p=fa[p];
31             }
32         }
33     }
34     int Query(char s[])
35     {
36         int len=strlen(s),ans=0,x=1;
37         for (int i=0; i<len; ++i)
38         {
39             if (!son[x][v[s[i]]]) return ans;
40             ++ans; x=son[x][v[s[i]]];
41         }
42         return ans;
43     }
44 }SAM;
45 
46 int main()
47 {
48     v['E']=0; v['S']=1; v['W']=2; v['N']=3;
49     scanf("%d%d%s",&n,&m,s);
50     for (int i=0; i<n; ++i) SAM.Insert(v[s[i]]);
51     for (int i=1; i<=m; ++i) scanf("%s",s), printf("%d
",SAM.Query(s));
52 }