HDU6153 A Secret 扩展KMP

题目链接：点击打开链接

这个题我还没想的很清楚，所以用网上的一个大佬的题解。等我想明白了，会把更加详细的解释补上的。

大佬的题解：点击打开链接

题目意思

给你两个字符串A,B，现在要你求B串的后缀在A串中出现的次数和后缀长度的乘积和为多少。

解题思路

扩展KMP模板题，将s和t串都逆序以后就变成了求前缀的问题了，扩展KMP求处从i位置开始的最长公共前缀存于数组，最后通过将数组的值不为0的进行一个等差数列和的和就可以了。

AC代码：

/*2017年8月20日10:48:30HDU6154还没有想明白 AC */ #include <iostream>#include <map>#include <set>#include <string>#include <cstring>#include <cstdio>#include <algorithm>#include <cmath>#include <queue>using namespace std;typedef long long ll;const int mod=1e9+7;const int maxn=1e6+10;   //字符串长度最大值  char s1[maxn],s2[maxn];int Next[maxn],ex[maxn]; //ex数组即为extend数组  //预处理计算next数组  //这部分为啥要这么算呢？/*因为每一个字符对答案的贡献是其长度乘以出现次数，从最长的字符串开始匹配，如果匹配成功一次，那么假设其长度为ll  出现一次l-1 出现一次..1 出现一次那么这个字符串长度的贡献即为这些求和。 */ ll add(ll n){ll m=((n%mod)*((n+1)%mod)/2)%mod; return m;}void getnext(char P[]){    int m=strlen(P);    Next[0]=m;    int j=0,k=1;    while(j+1<m&&P[j]==P[j+1]) j++;    Next[1]=j;    for(int i=2; i<m; 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<m&&P[i+j]==P[j])                j++;            Next[i]=j;            k=i;        }    }}void exkmp(char P[],char T[]){    int m=strlen(P),n=strlen(T);    getnext(P);    int j=0,k=0;    while(j<n&&j<m&&P[j]==T[j])        j++;    ex[0]=j;    for(int i=1; i<n; 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<n&&j<m&&T[i+j]==P[j])                j++;            ex[i]=j;            k=i;        }    }}int main(){int t;scanf("%d",&t);while(t--){scanf("%s%s",s1,s2);int len1=strlen(s1);int len2=strlen(s2);reverse(s1,s1+len1);reverse(s2,s2+len2);memset(Next,0,sizeof(Next));memset(ex,0,sizeof(ex));//getnext(s2);exkmp(s2,s1);/*for(int i=0;i<len1;i++) printf("ex[%d]=%d\n",i,ex[i]);for(int i=0;i<len2;i++){printf("Next[%d]=%d\n",i,Next[i]);*/ //}ll ans=0;for(int i=0;i<len1;i++){if(ex[i])ans=(ans+add(ex[i])%mod)%mod;}printf("%lld\n",ans%mod);} return 0;}