BZOJ1009 GT考试 （DP 矩阵乘法优化）

题目大意

阿申准备报名参加GT考试，准考证号为N位数X1X2….Xn(0<=Xi<=9),他不希望准考证号上出现不吉利的数字。

他的不吉利数学A1A2…Am(0<=Ai<=9)有M位，不出现是指X1X2…Xn中没有恰好一段等于A1A2…Am. A1和X1可以为
0

---

题解

DP，设状态f[i][j]为：长度为i的串 的后缀 至多与不吉利数字的前j位匹配 的串的个数。

则对于每一个状态j，加上相同的数字ch都会转移到相同的下一个状态j’，所以可以使用矩阵乘法加速dp。

详解见：http://blog.csdn.net/cjk_cjk/article/details/43038377

---

代码：

#include <cstdio>
#include <iostream>
using namespace std;

int moder;
int n,m;
char s[30];
inline int add(const int &a,const int &b) {
    register int c=a+b;
    return c>=moder?c-moder:c;
}
inline int mul(const int &a,const int &b) {
    register int c=a*b;
    return c%moder;
}

struct Matrix {
    int v[30][30];
    int n;
    Matrix(int id=1) {
        n=m;
        for(int i=0;i<30;i++)
            for(int j=0;j<30;j++)
                v[i][j]=0;
        for(int i=0;i<30;i++) v[i][i]=id;
    }
    void print() {
        for(int i=0;i<n;i++) {
            for(int j=0;j<n;j++) cout<<v[i][j]<<" ";
            cout<<endl;
        }
    }
    Matrix operator * (const Matrix &b) const {
        Matrix R; R.n=n;
        for(int i=0;i<n;i++)
        for(int j=0;j<n;j++) {
            int sum=0;
            for(int k=0;k<n;k++) sum=add(sum,mul(v[i][k],b.v[k][j]));
            R.v[i][j]=sum;
        }
        return R;
    }
}shift(0),res(0);

void construct(int id,int *v) {
    char a[30]; int pos;
    for(int i=1;i<=id;i++) a[i]=s[i];
    for(char ch='0';ch<='9';ch++) {
        a[pos=id+1]=ch;
        for(int len=pos;len>=0;len--) {
            if(!len) {v[len]++; break;}
            bool flag=true;
            for(int i=pos;i>=pos-len+1;i--) if(a[i]!=s[len-pos+i]) {
                flag=false;
                break;
            }
            if(flag) {v[len]++; break;}
        }
    }
}

Matrix fstpow(Matrix a,int k) {
    Matrix val(1); val.n=a.n;
    for(;k;k>>=1,a=a*a) if(k&1) val=val*a;
    return val;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
#endif // ONLINE_JUDGE
    scanf("%d%d%d\n",&n,&m,&moder);
    for(int i=1;i<=m;i++) scanf("%c",&s[i]);

    shift.n=m;
    for(int i=0;i<m;i++) construct(i,shift.v[i]);

    res.n=m;
    res.v[0][0]=1;
    res=res*fstpow(shift,n);

    int ans=0;
    for(int i=0;i<m;i++) ans=add(ans,res.v[0][i]);
    printf("%d\n",ans);

    return 0;
}