HihoCoder - 1807:好的数字串 (KMP DP)

Sample Input

6 1212

Sample Output

298

给定一个数字字符串S，如果一个数字字符串（只包含0-9，可以有前导0）中出现且只出现1次S，我们就称这个字符串是好的。

例如假设S=666，则1666、03660666是好的，6666、66、123不是好的；假设S=1212，则01212、12123是好的，121212、121是不好的。

请你计算长度为N的数字字符串中，有多少个是好的。由于总数可能很大，你只需要输出总数模1000000007的余数。

Input

一个整数N和一个数字串S。

对于30%的数据，1 ≤ N ≤ 8

对于100%的数据，1 ≤ N ≤ 1000，1 ≤ |S| ≤ N。

Output

一个整数代表答案。

思路：dp[i][j][k]，表示前i位，而且当前尾巴与S的前缀的最长公共长度位j，k表示是否出现过S。 对于第i位，枚举0到9，并且用KMP找到最长公共部分。

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=;
const int Mod=1e9+;
int a[maxn],dp[maxn][maxn][],ans,N,L;
int Next[maxn]; char c[maxn];
void KMP()
{
    for(int i=,k=;i<=L;i++){
        while(k&&a[i]!=a[k+]) k=Next[k];
        if(a[i]==a[k+]) k++; Next[i]=k;
    }
}
int get(int i,int j)
{
    while(){
        if(a[i+]==j) return i+;
        if(i==) break; i=Next[i];
    }return ;
}
int main()
{
    scanf("%d%s",&N,c+); L=strlen(c+); a[L+]=-;
    rep(i,,L) a[i]=c[i]-''; dp[][][]=;
    KMP();
    rep(i,,N){
        rep(j,,min(i-,L)){
            rep(k,,) {
                int t=get(j,k);
                if(j==L-){
                    if(t==L) (dp[i][t][]+=dp[i-][j][])%=Mod;
                    else {
                        (dp[i][t][]+=dp[i-][j][])%=Mod;
                        (dp[i][t][]+=dp[i-][j][])%=Mod;
                    }
                }
                else {
                    if(t!=L) (dp[i][t][]+=dp[i-][j][])%=Mod;
                    (dp[i][t][]+=dp[i-][j][])%=Mod;
                }
            }
        }

    }
    rep(i,,L) (ans+=dp[N][i][])%=Mod;
    printf("%d\n",ans);
    return ;
}