描述
题解
这个题不难解,因为是模版题,我有一个模版是专门查找模式串在原串中出现的次数,而这个题呢?求的是每一个模式串的后缀串在原串中出现的次数乘以长度之和,所以一开始我是暴力枚举后缀的,然而超时了,后来发现这个模版的求解思路其实就是从局部到全局的一个查找,那么我们完全可以在一次查找中查找出所有的后缀出现的次数,当然,说起后缀可能有些绕,我们先将两个串全部翻转一下就好了,这样就成前缀串了。
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN = 1e6 + 10;
const int MOD = 1e9 + 7;
int nt[MAXN];
int val[MAXN];
char S1[MAXN], S2[MAXN];
void get_nt(char *p, int *nt, int *val)
{
nt[0] = -1;
nt[1] = 0;
val[0] = 0;
val[1] = 1;
int len = (int)strlen(p);
for (int i = 1; i < len; i++)
{
nt[i + 1] = 0;
val[i + 1] = i + 1;
for (int j = nt[i]; j >= 0; j = nt[j])
{
if (p[i] == p[j])
{
nt[i + 1] = j + 1;
val[i + 1] += val[j + 1];
val[i + 1] %= MOD;
break;
}
}
}
}
int KMP_Count(char *p, int *nt, char *t, int *val)
{
int len = (int)strlen(t);
int ans = 0;
int j = 0;
for (int i = 0; i < len; i++)
{
int k = j;
for (j = 0; k >= 0; k = nt[k])
{
if (t[i] == p[k])
{
j = k + 1;
break;
}
}
ans = (ans + val[j]) % MOD;
}
return ans;
}
int main()
{
int T;
scanf("%d", &T);
while (T--)
{
scanf("%s%s", S1, S2);
int len_1 = (int)strlen(S1), len_2 = (int)strlen(S2);
reverse(S1, S1 + len_1);
reverse(S2, S2 + len_2);
get_nt(S2, nt, val);
printf("%d\n", KMP_Count(S2, nt, S1, val));
}
return 0;
}