BZOJ 1009--[HNOI2008]GT考试【KMP算法】【矩阵优化线性递推】

Description

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

题解

初看题目，感觉是组合数学的题目，但是并不可以。那就换一条思路。注意到M很小，想法肯定要往模式串上走，那么，什么算法和模式串有关呢？答案是KMP。定义$f[i][j]$表示匹配到第i位时匹配了前j位的方案数。则，转移方程显然是$f[i+1][k]+=f[i][j]$这样的形式，至于k是多少，只要枚举一下下一位的具体数字是多少，然后用KMP直接算出就可以。如果这样推，复杂度为O(10NM)，显然是不可行的，但是，这个递推式有一点特别的地方，只要模式串给定了，那么每一个状态由前面的那些状态累加而来就是确定而且完全可以算出的，那么，我们一定可以设计出这样一个矩阵A，使得

$$A*\begin{bmatrix}f[i][0]\\f[i][1]\\f[i][2]\\...\\f[i][M]\end{bmatrix}=\begin{bmatrix}f[i+1][0]\\f[i+1][1]\\f[i+1][2]\\...\\f[i+1][M]\end{bmatrix}$$ 我们事先构造出A数组，那么只要计算一次矩阵的快速幂就可以了，复杂度将变成logn级别的。

那么，如何来构造矩阵A呢？其实也很简单。显然，矩阵A是一个M*M的矩阵，只要枚举所有状态，如果当前状态可以由前一状态累加而来，那么矩阵的相应位置就加1（如，f[i][j]可以由f[i-1][k]累加而来，那么就在矩阵A的第i行第j列加一，注意，一定要累加而不是直接赋1，因为当前位上可以由多种数字，有可能有很多都能推到这个状态）。构造的代码如下：

ans.n=ans.m=m;
for(int i=0;i<m;i++)
 for(int j=0;j<=9;j++){
     int k=i;
     while(k&&a[k+1]!=j)k=p[k];
     if(a[k+1]==j)k++;
     if(k<m)ans.a[k][i]++;
 }

那么，A数组就构造好了，答案就是

$$A^N*\begin{bmatrix}1\\0\\0\\...\end{bmatrix}的0..m-1行的加和$$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 26
using namespace std;
int n,m,K,ans0,p[maxn],a[maxn];
struct matrix{
    int n,m,a[maxn][maxn];
    matrix operator *(matrix&b){
        matrix c;
        c.n=n;c.m=b.m;
        for(int i=0;i<=c.n;i++)
         for(int j=0;j<=c.m;j++){
            c.a[i][j]=0;
            for(int k=0;k<=m;k++)
             (c.a[i][j]+=a[i][k]*b.a[k][j])%=K;
         }
        return c;
    }
}ans,ans1;
matrix power(matrix x,int y){
    if(y==0){
        matrix c;
        c.n=c.m=m;
        for(int i=0;i<=m;i++)
         for(int j=0;j<=m;j++) if(i==j)c.a[i][j]=1;
                                  else c.a[i][j]=0;
        return c;
    }
    if(y==1)return x;
    matrix c=power(x,y>>1);
    if(y&1)return c*c*x;
      else return c*c;
}
int main(){
    freopen("gt.in","r",stdin);
    freopen("gt.out","w",stdout);
    scanf("%d%d%d",&n,&m,&K);getchar();
    for(int i=1;i<=m;i++)a[i]=getchar()-48;
    for(int i=2;i<=m;i++){
        int j=p[i-1];
        while(j&&a[j+1]!=a[i])j=p[j];
        if(a[j+1]==a[i])p[i]=j+1;
    }
    ans.n=ans.m=m;ans1.n=m;ans1.m=0;memset(ans1.a,0,sizeof(ans1.a));ans1.a[0][0]=1;
    for(int i=0;i<m;i++)
     for(int j=0;j<=9;j++){
        int k=i;
        while(k&&a[k+1]!=j)k=p[k];
        if(a[k+1]==j)k++;
        if(k<m)ans.a[k][i]++;
     }
    ans=power(ans,n)*ans1;
    for(int i=0;i<m;i++)(ans0+=ans.a[i][0])%=K;
    printf("%d",ans0);
    return 0;
}