JZOJ4886【NOIP2016提高A组集训第13场11.11】字符串

Description

某日mhy12345在教同学们写helloworld，要求同学们用程序输出一个给定长度的字符串，然而发现有些人输出了一些“危险”的东西，所以mhy12345想知道对于任意长度n的小写字母字符串，不包含危险串的字符串个数

Data Constraint

对于10%的数据，|str|=1
对于另30%的数据，n<=5
对于另30%的数据，危险串不存在相同字符
对于100%的数据，0<=|str|<=100,0<=n<=10000

Solution

这道题本人在考场已想出大半，但就是没想到怎样处理存在相同字符……，刚开始什么都没想到，只打了10分和30分的暴力，后来想到可以用dp解决剩下30分的部分分，于是没往下想就上了……

我们设出f[i][j]表示当前选到字符串的第i位，匹配到危险字符串的第j位的方案数，c[i][j]表示当前匹配到危险字符串的第i位，第i+1位为字母j时kmp中的失败指针的指向位置。那么对于一个f[i][j]，它的转移方程明显有$f[i+1][c[j][k]]+=f[i][j]$。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=205,maxn1=10005,mo=1e9+7;
ll n,i,t,j,k,l,len,num,ans,p[maxn],f[maxn1][maxn],c[maxn1][maxn];
char ch,s[maxn1],s1[maxn1];
bool bz;
int main(){
    freopen("helloworld.in","r",stdin);freopen("helloworld.out","w",stdout);
    //freopen("data.in","r",stdin);
    while (scanf("%lld",&n)!=EOF){
        scanf("\n");
        scanf("%s\n",s+1);
        bz=true;
        len=strlen(s+1);
            j=0;
            for (i=2;i<=len;i++){
                while (s[j+1]!=s[i] && j) j=p[j];
                if (s[j+1]==s[i]) j++;
                p[i]=j;
            }
            for (i=0;i<=len;i++)
                for (j=97;j<=122;j++){
                    k=i;
                    while (s[k+1]!=j && k) k=p[k];
                    if (s[k+1]==j) k++;
                    c[i][j]=k;
                }
            memset(f,0,sizeof(f));f[0][0]=1;
            for (i=0;i<n;i++)
                for (j=0;j<len;j++)
                    for (k=97;k<=122;k++)
                        f[i+1][c[j][k]]=(f[i+1][c[j][k]]+f[i][j])%mo;
            ans=0;
            for (j=0;j<len;j++)
                ans=(ans+f[n][j])%mo;
        printf("%lld\n",ans);
    }   
}