题目链接:点击打开链接
这个题我还没想的很清楚,所以用网上的一个大佬的题解。等我想明白了,会把更加详细的解释补上的。
大佬的题解:点击打开链接
题目意思
给你两个字符串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;
}