题目
某日mhy12345在教同学们写helloworld,要求同学们用程序输出一个给定长度的字符串,然而发现有些人输出了一些“危险”的东西,所以mhy12345想知道对于任意长度n的小写字母字符串,不包含危险串的字符串个数。
0<=|str|<=100,0<=n<=10000
时间限制 1s
空间限制 256M
解题思路
DP。f[i][j]表示到第i位,且前i位的后缀恰好是危险串的前j位。怎么转移呢?
f[i+1][pr[j][k]]=f[i+1][pr[j][k]]+f[i][j],pr[j][k]表示在危险串的第j位后接上第k个字母,这个串的前缀和后缀最多可以匹配多少位,说白了就是个类似kmp的东西。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 10006
#define maxl 106
#define fr(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const ll ding=1e9+7;
int i,j,k,n,len,p[maxl],pr[maxl][28];
ll ans,f[maxn][maxl];
char st[maxl];
void pre()
{
int i,j=0,k;
fr(i,2,len)
{
while (j && st[j+1]!=st[i]) j=p[j];
if (st[i]==st[j+1]) ++j;
p[i]=j;
}
fr(i,0,len)
fr(j,97,122)
{
k=i;
while (k && st[k+1]!=j) k=p[k];
if (st[k+1]==j) ++k;
pr[i][j-96]=k;
}
return;
}
int main()
{
freopen("helloworld.in","r",stdin);
freopen("helloworld.out","w",stdout);
while (scanf("%d",&n)!=EOF)
{
scanf("%s",st+1);
len=strlen(st+1);
pre();
memset(f,0,sizeof(f));
f[0][0]=1;
fr(i,0,n-1)
fr(j,0,len-1)
fr(k,1,26)
f[i+1][pr[j][k]]=(f[i+1][pr[j][k]]+f[i][j])%ding;
ans=0;
fr(i,0,len-1) ans=(ans+f[n][i])%ding;
printf("%lld\n",ans);
}
return 0;
}