题目链接:点击打开链接
给你字符串的长度以及这个字符串,问你前缀数量之和是多少。
利用kmp中next数组的性质,将该跳多少位保存到nxt数组中,dp数组保存当前子串i所含前缀数量之和,dp[i] = (dp[nxt[i]] + 1) % MOD
计算当前结尾子串i数量加上前nxt[i]个字符这一前缀的数量。
AC代码:
#include "iostream"
#include "cstdio"
#include "cstring"
#include "algorithm"
using namespace std;
const int MAXN = 2e5 + 5;
const int MOD = 10007;
char s[MAXN];
int len, nxt[MAXN], dp[MAXN];
void get_nxt()
{
nxt[1] = 0;
int j = 0;
for(int i = 2; i <= len; ++i) {
while(j > 0 && s[j + 1] != s[i])
j = nxt[j];
if(s[j + 1] == s[i]) j++;
nxt[i] = j;
}
}
int main(int argc, char const *argv[])
{
int t;
scanf("%d", &t);
while(t--) {
memset(nxt, 0, sizeof(nxt));
memset(dp, 0, sizeof(dp));
scanf("%d", &len);
scanf("%s", s + 1);
get_nxt();
int ans = 0;
for(int i = 1; i <= len; ++i) {
dp[i] = (dp[nxt[i]] + 1) % MOD;
ans = (ans + dp[i]) % MOD;
}
printf("%d\n", ans);
}
return 0;
}