題目大意:
給你兩個字符串A,B,現在要你求B串的后綴在A串中出現的次數和后綴長度的乘積和為多少。
題解:
擴展KMP模板題,將A和B串都逆序以后就變成了求前綴的問題了,擴展KMP求處從i位置開始的最長公共前綴存于數組。
最后通過將數組的值不為0的進行一個桶計數,倒著進行一下求和就可以了。注意,在這個題目上擴展kmp處理出來的是
ex[ i ]數組是 A串的每個從 i 位置開始的后綴 ,與B串的最長公共前綴長度,那么這樣B串在A串上匹配的情況就一目了然了。
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int N = 1000005;
int Next[N];
long long ex[N],tong[N]; //即extand[]
char p[N],t[N];
int T;
long long ans;
void pre() // next[i]: 以第i位置開始的子串 與 T的公共前綴
{int lp=strlen(p);Next[0]=lp;int j=0,k=1;while(j+1<lp && p[j]==p[j+1]) j++;Next[1]=j;for(int i=2; i<lp; i++){int P=Next[k]+k-1;int L=Next[i-k];if(i+L<P+1) Next[i]=L;else{j=max(0,P-i+1);while(i+j<lp && p[i+j]==p[j]) j++; // 枚舉(p+1,length) 與(p-k+1,length) 區間比較Next[i]=j;k=i;}}
}
void exkmp()
{int lp=strlen(p),lt=strlen(t);pre(); //next數組初始化int j=0,k=0;while(j<lt && j<lp && p[j]==t[j]) j++;ex[0]=j;for(int i=1; i<lt; i++){int P=ex[k]+k-1;int L=Next[i-k];if(i+L<P+1) ex[i]=L;else{j=max(0,P-i+1);while(i+j<lt && j<lp && t[i+j]==p[j]) j++;ex[i]=j;k=i;}}
}
int main()
{scanf("%d",&T);while(T--){memset(tong,0,sizeof(tong));scanf("%s%s",&t,&p);int lt=strlen(t);int lp=strlen(p);reverse(p,p+lp);reverse(t,t+lt);exkmp();ans=0;for(int i=0; i<lt; i++)tong[ex[i]]++;for(int i=lp;i;i--){tong[i]=(tong[i]+tong[i+1])%mod;ans=(ans+tong[i]*i%mod)%mod;}printf("%lld\n",ans);}
}