原题:https://www.lydsy.com/JudgeOnline/problem.php?id=4327
题解:将模式串加入ac自动机跑主串。对于每一个前缀,打上标记跳失败指针,统计答案即可。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
const int N=1e7+10,MAXL=110,Z=4,M=1e5+10;
map<char,int> temp;
queue<int> q;
int ch[N][Z],end[M],len[M],nxt[N],fa[N],n,m,tot;
bool f[N]={0};
char s[N],t[MAXL];
inline void init(){
temp['N']=0;temp['S']=1;temp['W']=2;temp['E']=3;
tot=1;f[0]=f[1]=1;//终点
}
inline void ins(char *s,int ind){
len[ind]=strlen(s+1);int p=1;
for(int i=1,c;i<=len[ind];i++){
c=temp[s[i]];
if(!ch[p][c]) ch[p][c]=++tot;
fa[ch[p][c]]=p;//父节点
p=ch[p][c];
}
end[ind]=p;
}
inline void pre(){
for(int i=0;i<Z;i++) ch[0][i]=1;//失败指针
q.push(1);nxt[1]=0;
while(!q.empty()) {
int u=q.front();q.pop();
for(int i=0;i<Z;i++){
if(!ch[u][i]) ch[u][i]=ch[nxt[u]][i];
else {
q.push(ch[u][i]);
int v=nxt[u];
while(v>0 && !ch[v][i]) v=nxt[v];
nxt[ch[u][i]]=ch[v][i];
}
}
}
int p=1;
for(int i=1,c;i<=n;i++){//跑主串,打标记
c=temp[s[i]];
for(int k=ch[p][c];k>1&&!f[k];k=nxt[k]) f[k]=1;
p=ch[p][c];
}
}
int main(){
// freopen("bzoj4327.in","r",stdin);
init();
scanf("%d%d%s",&n,&m,s+1);
for(int i=1;i<=m;i++){
scanf("%s",t+1);ins(t,i);
}
pre();
for(int i=1;i<=m;i++){
for(int k=end[i];k && !f[k];k=fa[k]) len[i]--;//统计答案
printf("%d\n",len[i]);
}
return 0;
}