题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=3336
题目意思:
给你一个字符串,要求求出所求的前缀在字符串一共出现了多少次
解题思路:
KMP+DP
我们令dp[i]表示以s[i]结尾的字符串中出现了几个以s[i]结尾的前缀
那么对于每个i而言,我们就是找出在1~i-1里面的和s[i]相匹配的前缀j
得到dp[i]=dp[j]+1(s[i]==s[j])
如果找到的这个不相等的话,那么就说明之前根本就么有
则dp[i]=1;
下面上代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 200000+10;
char s[maxn];
int dp[maxn];
int f[maxn];
int ans;
int len;
void getfail()
{
int m=len;
f[0]=0;
f[1]=0;
for(int i=1;i<m;i++)
{
int j=f[i];
while(j && s[i]!=s[j])
j=f[j];
f[i+1]= (s[i]==s[j])?j+1:0;
}
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d",&len);
scanf("%s",s);
ans=1;
getfail();
dp[0]=1;
int j=0;
for(int i=1;i<len;i++)
{
j=f[i];
while(j>0 && s[i]!=s[j])
j=f[j];
if(s[j]==s[i])
dp[i]=dp[j]+1;
else
dp[i]=1;
ans=(ans+dp[i])%10007;
}
printf("%d\n",ans);
}
return 0;
}