题意:给一个字符串,输出包括从1到len长 字符串前缀的总个数
YY: 基本思想是 KMP + DP,主要看KMP
kmp思想:对字符串进行预处理,记录与当前位置i后缀相同的“最近”位置,用next[i]记录,
保证 s[1 .. i] 中 s[i - next[i] + 1 .. i] 与 s[1 .. next[i]] 是相同的,
以便在某处字符不匹配时,不用重新从头判断一遍,只要从对应的next[i]即可,
因为中间有部分与自身重叠,减少了不必要的判断,实现见代码。
例1:
i : 1234567
字符串 : abcabab
next[i] : 0001212
例2:
i : 12345678
字符串 : abababab
next[i] : 00123456
例3:
i : 123456789
字符串: ababcdabd
next[i]: 001200120
DP:
cns[]数组:以 i 结尾的串中所有前缀的计数和
状态转移方程: cns[i] = cns[next[i]] + 1;
i : 123456
字符串: ababab
cns[i]: 112233
比如当i等于5时,ababa中后面的aba与前面的aba重叠,
以第三个a为结尾的前缀总数对应于前面的aba的前缀数+1(ababa)
<pre name="code" class="cpp">#include <stdio.h>
#include <string.h>
const int MAX_LEN = 200005;
int next[MAX_LEN];
char s[MAX_LEN];
int cns[MAX_LEN];
int main()
{
int cas;
scanf("%d",&cas);
while(cas--)
{
int len;
int flag;
scanf("%d",&len);
scanf("%s",s+1); //!!!!!!!!!!!!!
next[1] = 0;
flag = 0;
for(int i=2;i<=len;i++)
{
while(flag > 0 && s[flag+1]!=s[i])
flag = next[flag];
if(s[flag+1] == s[i])
flag++;
next[i] = flag;
}
int sum=0;
memset(cns,0,sizeof(cns));
for(int i=1;i<=len;i++)
{
cns[i] = (cns[next[i]] + 1)%10007;
sum = (sum+cns[i])%10007;
}
printf("%d\n",sum);
}
return 0;
}