题目大意:
给你两个字符串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 ",ans); } }