A - A Secret -扩展KMP

题目大意：

给你两个字符串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);

    }
}