HDU 6153 A Secret 扩展KMP

题意
在母串中找模式串的所有后缀的匹配次数，并乘上各种后缀对应长度，再把所有的结果加起来%1e9+7输出。
思路：
把两个字符串都反转一下就变成了前缀匹配，就可以利用扩展KMP的性质了。因为extend[i] 表示的是从i到结尾的主串与模式串的最长公共前缀，这样遍历一遍主串，每一位对答案的贡献就是1+⋯+extend[i]=extend[i]×(extend[i]+1)/2，累加即可。
举个栗子 （下面是反转后的字符串）
S串： abbabbab
T串： abbc
extend[3]=3,那么长度为1,2,3的“a”,“ab”,"abb"各匹配一次，贡献为(1+2+3),即(1+2+…+extend[i])，即(extend[i]*(extend[i]+1))/2。

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<stack>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e6+5;
const int MOD=1e9+7;
int nex[maxn];
int extend[maxn];
void pre_exkmp(char *b)
{
    int len=strlen(b);
    nex[0]=len;
    int j=0;
    while(j+1<len&&b[j]==b[j+1]) j++;
    nex[1]=j;
    int k=1;
    for(int i=2;i<len;i++)
    {
        int p=nex[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) nex[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<len&&b[i+j]==b[j]) j++;
            nex[i]=j;
            k=i;
        }
    }
}
void exkmp(char *a,char *b)
{
    pre_exkmp(b);
    int j=0;
    int lena=strlen(a);
    int lenb=strlen(b);
    while(j<lena&&j<lenb&&a[j]==b[j]) j++;
    extend[0]=j;
    int k=0;
    for(int i=1;i<lena;i++)
    {
        int p=extend[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) extend[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lena&&j<lenb&&a[i+j]==b[j]) j++;
            extend[i]=j;
            k=i;
        }
    }
}
char a[maxn];
char b[maxn];
int main()
{
    
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s%s",a,b);
        int lena=strlen(a);
        int lenb=strlen(b);
        strrev(a);
        strrev(b);
        exkmp(a,b);
        ll sum=0;
        for(int i=0;i<lena;i++)
            sum=(sum+((long long)extend[i]*(extend[i]+1)/2)%MOD)%MOD;
        printf("%lld\n",sum);
    }
    return 0;
}