HDU6153 A Secret 扩展KMP

题目链接：点击打开链接

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

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

题目意思

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

解题思路

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

AC代码：

/*
2017年8月20日10:48:30
HDU6154
还没有想明白 
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数组  

//这部分为啥要这么算呢？
/*
因为每一个字符对答案的贡献是其长度乘以出现次数，
从最长的字符串开始匹配，如果匹配成功一次，那么假设其长度为l
l  出现一次
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;
}