BZOJ 1009 KMP思想 + DP + 矩阵快速幂

第一次用MarkDown和LaTex，写得有点丑……

本题的坑爹历程给了我一个血的教训：没有真正搞清楚做法之前，不要瞎BB地写题解。不然会造成深陷坑中的严重后果。

题意简述：给定一个字符串s，求出长度为n的不含字串s的字符串t的数量。

这道题是一个非常经典的模型，DP之：
设 f[i][j]为前i个t字符，匹配到s的第 j位（强制选 i）的方案数，则有

ans=Σ( f[n][k] | 0≤k<m )

接下来考虑 f[i][j]的转移：
由于我们是要统计数量，所以我们可以枚举第 i+1位是几。然后自然地，应该转移到（这里亦即贡献到）f[i+1][pos]，其中pos为加上这个字符之后，当前串匹配到的位置。

然后我们发现，每一轮i=>i+1的转移都是一样的。对于这样的“无脑”递推，我们可以构造矩阵，然后使用矩阵快速幂来加速。如何构造矩阵呢？我们考虑矩阵元素M[i][j]的意义。它会与当前向量的第j个维度相乘，然后贡献到下一个向量的第i个维度。因此，如果有f[i+1][pos] += f[i][j]，就应把M[pos][j]加1。不难发现，处理好这个矩阵之后，它的第0列其实就是初始向量。所以，只需做n次矩阵乘法即可。

代码略丑：

// BZOJ 1009 KMP+Matrix

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

 const int M=25, N=M;

 #define rep(i,a,b) for (int i=a; i<=b; i++)
 #define dep(i,a,b) for (int i=a; i>=b; i--)
 #define read(x) scanf("%d", &x)
 #define fill(a,x) memset(a, x, sizeof(a))

 int mod;
 struct Matrix {
    int n, m, a[N][M];

    void init(int n, int m) {
        this->n=n;
        this->m=m;
        fill(a, 0);
    }

    Matrix operator * (const Matrix B) const {  
        Matrix ret;
        ret.init(B.n, B.n);
        rep(i,0,n-1) rep(j,0,n-1) rep(k,0,n-1) ret.a[i][j]=(ret.a[i][j]+a[i][k]*B.a[k][j])%mod;
        return ret; 
    }

    Matrix operator ^ (const int k) const {
        Matrix ret, t=*this;
        ret.n=n; ret.m=m;
        int tk=k;
        fill(ret.a, 0);
        rep(i,0,ret.n-1) ret.a[i][i]=1;
        while (tk) {
            if (tk&1) ret=ret*t;
            t=t*t;
            tk>>=1;     
        }
        return ret;
    }
 } A;

 int f[M], n, m;
 char st[M];

 void get_fail() {
    f[0]=f[1]=0;
    int j=0;
    rep(i,2,m) {
        while (j && st[j+1]!=st[i]) j=f[j];
        if (st[j+1]==st[i]) j++;
        f[i]=j;
    }
 }

int main()
{
    read(n); read(m); read(mod);
    scanf("%s", st+1);
    A.init(m, m);

    get_fail();

    rep(i,0,m-1) 
      rep(j,0,9) {
        int pos=i; 
        while (pos && st[pos+1]-'0'!=j) pos=f[pos];
        if (st[pos+1]-'0'==j) pos++;
        if (pos!=m) A.a[pos][i]=(A.a[pos][i]+1)%mod;
      }

    A=A^n;
    int ans=0;
    rep(i,0,m-1) ans=(ans+A.a[i][0])%mod;
    printf("%d
", ans);

    return 0;
}