HDU 6153 A Secret（扩展kmp模板题）

给你两个字符串，求B的每一个后缀在A中出现的次数与其长度之积。

可以使用extend kmp求得ex【i】（表示对于B与subA（i，strlen（A）-1）的最大匹配前缀）；所以我们先反置两个字符串，之后，对于每一个得到的ex【i】，他可以带来的变化是（1+2+3+4+……+i）；所以加一个求和函数来得到答案，注意我代码中的ex【i】为int型，在Add函数中，我们如果还是传入int型的话，在运算中会炸。。。转化成longlong就行了

#include<bits/stdc++.h>
using namespace std;

#define maxn 1000001
#define mod 1000000007
typedef long long LL;
LL ans;

int t;
char str1[maxn],str2[maxn];
int Next[maxn],ex[maxn];

void getNext(char str[])
{
int i = 0, j, po, len = strlen(str);
Next[0] = len;
while(str[i] == str[i+1]
&&i + 1 < len) i++;
Next[1] = i; po = 1;
for(i = 2; i < len; i++)
{
//
if(Next[i - po] + i < Next[po] + po)
Next[i] = Next[i - po];
else
{
j = Next[po] + po - i;
if(j < 0) j = 0;
while(i + j < len&&str[j] == str[j + i])
j++;
Next[i] = j;
po = i;
}
}
}

void ex_kmp(char str1[], char str2[])
{
int i = 0, j, po, len1 = strlen(str1);
int len2 = strlen(str2);
getNext(str2);
while(str1[i] == str2[i]&&
i < len2&&i < len1) i++;
ex[0] = i;
po = 0;
for(int i = 1; i < len1; i++)
{
if(Next[i - po] + i < ex[po] + po)
ex[i] = Next[i - po];
else
{
j = ex[po] + po - i;
if(j < 0) j = 0;
while(i + j < len1&&j < len2
&&str1[i + j] == str2[j]) j++;
ex[i] = j;
po = i;
}
}
}

LL add(LL x)
{
return (x*(x+1))/2%mod;
}

int main()
{
scanf("%d",&t);
while(t--)
{
ans = 0;
scanf("%s%s", str1, str2);
int len1 = strlen(str1);
int len2 = strlen(str2);
reverse(str1,str1+len1);
reverse(str2,str2+len2);
ex_kmp(str1, str2);
for(int i = 0; i < len1; i++)
{
ans = ans + add(ex[i]);
ans %= mod;
}
printf("%lld\n", ans);
}
}