JZOJ 4886 字符串

题目

某日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;
}