【HNOI2008】GT考试

Description

申准备报名参加GT考试，准考证号为n位数X1X2X3…Xn-1Xn(0<=Xi<=9)，他不希望准考证号上出现不吉利的数字。
他的不吉利数字A1A2A3…Am-1Am(0<=Ai<=9)有m位，不出现是指X1X2X3…Xn-1Xn 中没有恰好一段等于A1A2A3…Am-1Am。A1和X1可以为0。
阿申想知道不出现不吉利数字的号码有多少种，输出模K取余的结果。

Input

第一行输入n,m,K，接下来一行输入m 位的数。

Output

模K取余的结果。

Sample Input

样例一：

4 3 100
111

样例二：
3 0 100

Sample Output

样例一：

81

样例二：

0

Data Constraint

对100%的数据满足n<=10^9,m<=20,K<=1000；

Solution

先设DP
设f[i,j]表示匹配到了上面第i位，下面匹配到了第j个数字（上面是指那个n位数，下面是指给定的m位数）

那么再枚举一个第i位数字k(0~9)然后转移
因为f[i,j]不一定只从f[i-1,j-1]转移过来，还可以从f[i-1,j+……]转移过来，为什么呢？想象成字符串匹配，再想一下KMP中的next数组，应该就可以理解了
那么转移显然，初始状态为f[0,0]=1

然后上矩阵乘法
用矩阵A[j,k]表示从f[i,j]可以转移到f[i+1,k]的次数（因为第i+1位可能是0~9），当k不是0时，A[j,k]一定是1，而当k为0时，就可能有多种了
根据矩阵乘法，当一维矩阵B[m]为f[i,0~m]时*矩阵A，就变成了f[i+1,0~m]。（手推一下）
然后按照套路矩阵快速幂
转移矩阵A是m*m的，时间复杂度为$O(log_2n*m^3)$

Code

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 25
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
int a[N][N],b[N][N],c[N][N],mo,s[N],next[N],m;
void gn()
{
    int j=0;
    fo(i,2,m)
    {
        while(s[i]!=s[j+1]&&j>0) j=next[j];
        if(s[i]==s[j+1]) j++;
        next[i]=j;
    }
}
void cl()
{
    fo(i,0,m) fo(j,0,m) c[i][j]=a[i][j],a[i][j]=0;
}
void ch1()
{
    cl();
    fo(i,0,m) fo(j,0,m) fo(k,0,m) a[i][k]=(a[i][k]+c[i][j]*c[j][k])%mo;
}
void ch2()
{
    cl();
    fo(i,0,m) fo(j,0,m) fo(k,0,m) a[i][k]=(a[i][k]+b[i][j]*c[j][k])%mo;
}
void mi(int n)
{
    if(n<=1) return;
    fo(i,0,m) fo(j,0,m) b[i][j]=a[i][j];
    mi(n/2);
    ch1();
    if(n%2==1) ch2();
}
int main()
{
    int n;scanf("%d%d%d\n",&n,&m,&mo);
    fo(i,1,m) scanf("%c",&s[i]),s[i]-=48;
    gn();next[0]=-1;m--;
    fo(i,0,m)
        fo(k,0,9)
        {
            int bz=1;
            for(int j=i;j>=0;j=next[j]) 
            if(s[j+1]==k) 
            {
                a[i][j+1]++;bz=0;break;
            }
            if(bz) a[i][0]++;
        }
    mi(n);
    memset(b,0,sizeof(b));
    b[0][0]=1;
    ch2();
    int ans=0;fo(i,0,m) ans=(ans+a[0][i])%mo;
    printf("%d",ans);
}