bzoj1009 [HNOI2008] GT考试矩阵乘法+dp+kmp

1009: [HNOI2008]GT考试

Time Limit: 1 Sec Memory Limit: 162 MB
Submit: 4542 Solved: 2815
[Submit][Status][Discuss]

Description

　　阿申准备报名参加GT考试，准考证号为N位数X1X2....Xn(0<=Xi<=9),他不希望准考证号上出现不吉利的数字。
他的不吉利数学A1A2...Am(0<=Ai<=9)有M位，不出现是指X1X2...Xn中没有恰好一段等于A1A2...Am. A1和X1可以为
0

Input

　　第一行输入N,M,K.接下来一行输入M位的数。 N<=10^9,M<=20,K<=1000

Output

　　阿申想知道不出现不吉利数字的号码有多少种，输出模K取余的结果.

Sample Input

4 3 100
111

Sample Output

HINT

矩阵乘法的题题解写起来都十分麻烦。。

而且很多东西只能意会。。

f[i , j]表示前 i 个准考证号匹配到不吉利串第 j 个的方案

然后你需要把一个答案矩阵f[i , j]转移到f[i+1 , j]

举个例子，样例，比如当前匹配到了第2位，也就是说前 i 位的结尾是11

对于第 i+1 个字符，如果是 1 的话，接着匹配到不吉利串第 3 位，不是 1 的话就匹配到第 0 位了

也就是说前 i 位匹配到了不吉利串 j 位，加入 i+1 这个字符，有不同情况，有一些会转移到j+1，一些会转移到其他的，写成一些形如f[i+1 , k] += f[i , j]的式子……

f[i+1 , 3] += f[i , 2]

f[i+1 , 0] += f[i , 2]

即枚举i+1可能出现的字符，然后看n个f[i , j]分别转移到哪去，就在转移矩阵的这个转移路径上+1

按照这个思路用kmp写出转移矩阵，事实上暴力应该就行了

 #include<iostream>

 #include<cstdio>

 #include<cstring>

 using namespace std;

 inline int read()

 {

     char ch=getchar();

     int f=,x=;

     while(!(ch>=''&&ch<='')){if(ch=='-')f=-;ch=getchar();}

     while(ch>=''&&ch<=''){x=x*+(ch-'');ch=getchar();}

     return x*f;

 }

 int n,m,mod;

 int p[];

 char ch[];

 int a[][],b[][];

 void mul(int a[][],int b[][],int ans[][])

 {

     int tmp[][];

     for(int i=;i<m;i++)

         for(int j=;j<m;j++)

         {

             tmp[i][j]=;

             for(int k=;k<m;k++)

                 tmp[i][j]=(tmp[i][j]+a[i][k]*b[k][j])%mod;

         }

     for(int i=;i<m;i++)

         for(int j=;j<m;j++)

             ans[i][j]=tmp[i][j];

 }

 int main()

 {

     n=read();m=read();mod=read();

     scanf("%s",ch+);

     int j=;

     for(int i=;i<=m;i++)

     {

         while(j>&&ch[j+]!=ch[i])j=p[j];

         if(ch[j+]==ch[i])j++;

         p[i]=j;

     }

     for(int i=;i<m;i++)

        for(int j=;j<=;j++)

        {

                int t=i;

             while(t>&&ch[t+]-''!=j)

                 t=p[t];

             if(ch[t+]-''==j)t++;

             if(t!=m)b[t][i]=(b[t][i]+)%mod;

        }

     for(int i=;i<m;i++)

         a[i][i]=;

     while(n)

     {

         if(n&)mul(a,b,a);

         mul(b,b,b);

         n>>=;

     }

     int sum=;

     for(int i=;i<m;i++)

         sum=(sum+a[i][])%mod;

     printf("%d",sum);

     return ;

 }