http://www.lydsy.com/JudgeOnline/problem.php?id=4327
对于小串建AC自动机
原串在AC自动机上匹配
对于一个点可以匹配到,那么它的fail指针也可以匹配到
做完以后,按bfs序倒着转移一下标记
如何对于小串进行查询
#include<cstdio> #include<queue> #define FOR(i,s,t) for(register int i=s;i<=t;++i) #define ROF(i,s,t) for(register int i=s;i>=t;--i) const int N=10000011,M=100011; char S[N]; char A[M][101]; int xu[N]; int n,m,tot; namespace AC_automaton{ int go[128]; struct Tire{ int fail; int vis; int to[4]; }tr[N]; inline void insert(char *s){ int u=0; for(register int i=0;s[i]!=‘\0‘;++i){ if(!tr[u].to[go[(int)s[i]]]) tr[u].to[go[(int)s[i]]]=++tot; u=tr[u].to[go[(int)s[i]]]; } } inline void build_fail(){ std::queue<int>q; int v,u,now; FOR(i,0,3) if((v=tr[0].to[i])) q.push(v); while(!q.empty()){ now=q.front(); xu[++xu[0]]=now; q.pop(); FOR(i,0,3){ v=tr[now].to[i]; u=tr[now].fail; if(v){ tr[v].fail=tr[u].to[i]; q.push(v); } else tr[now].to[i]=tr[u].to[i]; } } } inline void match(char *s){ int now=0; for(register int i=0;i<n;++i){ now=tr[now].to[go[(int)s[i]]]; tr[now].vis=1; } } inline int query(char *s){ int u=0,pos=0; for(register int i=0;s[i]!=‘\0‘;++i){ u=tr[u].to[go[(int)s[i]]]; if(tr[u].vis) pos=i+1; } return pos; } } using namespace AC_automaton; int main(){ go[‘E‘]=0;go[‘S‘]=1;go[‘W‘]=2;go[‘N‘]=3; scanf("%d%d\n",&n,&m); scanf("%s",S); FOR(i,1,m){ scanf("%s",A[i]); insert(A[i]); } build_fail(); match(S); ROF(i,xu[0],1)tr[tr[xu[i]].fail].vis|=tr[xu[i]].vis; FOR(i,1,m) printf("%d\n",query(A[i])); return 0; }