题意:给定一个字符串,对于该字符串的所有前缀,找出每个前缀在字符串中出现的次数,并把次数相加得出最终的结果。
解题思路:利用KMP的失配函数f[i],因为对于每个f[i]函数的对应值j来说,长度为j的字符串已经出现一次,长度为j-1的该字符也已经出现,故++case[j]记录第一次出现的情况,当j>0时,++case[j-1]记录j-1前缀超过2次的情况;依次类推就可以得出每个前缀在字符串中出现的总和。
Code:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 200009;
int next[maxn], count[maxn],n;
char pattn[maxn] ;
void getNext ()
{
int i = 0, j = -1 ;
next[0] = -1 ;
memset (count, 0, sizeof (count)) ;
while (i < n)
{
if (j >= 0 && pattn[i] != pattn[j]) j = next[j] ;
else
{
++count[i] ;
++i ; ++j ; next[i] = j;
if (j > 0) ++count[j - 1] ;
}
}
}
int main ()
{
int T, sum ;
while (~scanf ("%d", &T))
{
while (T--)
{
scanf ("%d", &n) ;
scanf ("%s", pattn) ;
getNext () ;
sum = 0 ;
for (int i = 0 ; i < n ; ++i)
sum += count[i] ;
printf ("%d\n", sum % 10007) ;
}
}
return 0 ;
}