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