BZOJ 4327: JSOI2012 玄武密码 后缀自动机
Code:
#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;
}