KMP的一道水题,最初竟然傻逼的想用自动机怼,毫无疑问的T了,后来仔细想了一下,就是自动自单链的情况,就怼kmp,然后不知道发什么神经没有转过来弯,竟然没调出来,然后我就自己写了单链的fail指针,调了很久A了,但是浪费了很多时间,赛后想想,特么的就是个kmp,next数组跟fail指针就特么差一位而已,果然,套个板子就A了,要不然,今天又可以多出一题了,警示自己一下,不要那么菜啊~~~~
#include <stdio.h> #include <algorithm> #include <iostream> #include <string.h> #include <cmath> #include <queue> using namespace std; const int maxn=1e6+10; const int mod=1e9+7; int Next[maxn],en[maxn]; void kmp_pre(char x[],int m) { int i,j; j=Next[0]=-1; i=0; while(i<m) { while(-1!=j && x[i]!=x[j])j=Next[j]; Next[++i]=++j; } } int KMP_Count(char x[],int m,char y[],int n) { memset(en,0,sizeof en); int j=0; int ans=0; kmp_pre(x,m); for(int i=0;i<n;i++) { if(x[j]==y[i]) { en[j]++;j++; if(j==m)j=Next[j]; continue; } j=Next[j]; while(j!=-1 && x[j]!=y[i])j=Next[j]; if(j==-1){j=0;continue;} en[j]++;j++; if(j==m)j=Next[j]; } for(int i=m;i>0;i--) en[Next[i]-1]+=en[i-1]; for(int i=0;i<m;i++) ans=(ans+1ll*en[i]*(i+1)%mod)%mod; return ans; } char s[maxn],ts[maxn]; int main() { freopen("input.txt","r",stdin); int T;scanf("%d",&T); while(T--) { scanf("%s%s",s,ts); strrev(s); strrev(ts); int ans=KMP_Count(ts,strlen(ts),s,strlen(s)); printf("%d\n",ans); } return 0; }