【HNOI2008】GT考试

题目

申准备报名参加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取余的结果。
对100%的数据满足n<=10^9,m<=20,K<=1000；

分析

矩阵乘法。
我们设f[i,j]为数字的前i位和匹配串的前j项匹配，
那么这样很明显确定了匹配串前j位了，就可以确定后一位即j+1，i+1位的转移了。
也就是我们确定了是和匹配串匹配j位，那么我们也可以把它可以从哪里转移找出来。
那么这样问题便变得很简单了。
最后的答案便是匹配0—m-1的方案的和

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int N=25;
typedef int martix[N][N];
int n,m,mo,t;
martix c,d,a;
char q[N],sum[N][N],Q[N];
void mul(martix a,const martix b){
    martix c;
    memset(c,0,sizeof(c));
    for(int i=0;i<=m-1;i++)
       for(int j=0;j<=m-1;j++) 
          for(int k=0;k<=m-1;k++)
             c[i][j]+=a[i][k]*b[k][j],c[i][j]%=mo;
    memcpy(a,c,sizeof(martix));
}
bool pd(char a[N],int l,char b[N],int k){
    for(int i=1;i<=k;i++)
    if (a[l-i+1]!=b[k-i+1]) return 0;
    return 1;
}
int main(){
    scanf("%d%d%d",&n,&m,&mo);
    scanf("%s",Q);
    for(int i=1;i<=m;i++) q[i]=Q[i-1];
    memset(sum,0,sizeof(sum));
    for(int i=1;i<=m;i++){
        for(int j=1;j<=i;j++) sum[i][j]=q[j];
    }
    a[0][1]=1;
    a[0][0]=9;
    for(int i=1;i<=m-1;i++){
        for(int j=0;j<=9;j++){
            if (i==m-1&&j==int(q[m]-48)) 
            continue;
            char NEW[N];
            for(int k=1;k<=i;k++) NEW[k]=sum[i][k];
            NEW[i+1]=char(j+48);int p=1;
            for(int k=i+1;k;k--){
                if (pd(NEW,i+1,sum[k],k)) {
                    a[i][k]++;p=0;break;
                }
            }
            if (p) ++a[i][0];
        }
    }
    for(int i=0;i<=m-1;i++) d[i][i]=1;
    while (n){
        if(n&1) mul(d,a);
        mul(a,a);
        n=n>>1;
    }
    int ans=0;
    for(int i=0;i<=m-1;i++)ans+=d[0][i],ans%=mo;
    printf("%d",ans);
}