bzoj1009: [HNOI2008]GT考试

矩阵快速幂。。。第一次做矩阵的快速幂 花了不少时间。

先假定f[i,j]表示到i为止，可以匹配j位。最终答案便是f[n,j](0<=j<m),没有完全匹配的.

由于n比较大，所以不可能直接求解。我们可以发现f[i,j]可以由f[i-1,k](0<=k<m)推出，之间可以求解出一个关系式，其中可以KMP求匹配。

通过该关系式 构造一个矩阵，然后矩阵快速幂就可以迅速求解了。

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cstring>
#include <cmath>
using namespace std;
int next[23], b[21], c[21];
struct REC
{
    int a[21][21];//新长度，旧长度
    REC(){
        memset(a,0,sizeof(a));
    }
} r;
char s[23];
void kmp()
{
    int i, j, len;
    len=strlen(s);
    next[0]=-1, i=0, j=-1;
    while(i<len){
        if(j==-1||s[i]==s[j]){
            i++, j++;
            next[i]=j;
        }else j=next[j];
    }
    int t;
    for(i=0;i<len;i++){
        for(j=0;j<=9;j++){
            t=i;
            while(t!=-1&&j+'0'!=s[t]) t=next[t];
            t++;
            r.a[t][i]++;
        }
    }
}
int main()
{
    int i, j, e, n, m, k, len;
    while(~scanf("%d%d%d%s",&n,&m,&k,s)){
        kmp();
        len=strlen(s);
        memset(b,0,sizeof(b));
        b[0]=1;
        while(n){
            if(n%2){
                memset(c,0,sizeof(c));
                for(i=0;i<len;i++) for(j=0;j<len;j++){
                        c[i]+=r.a[i][j]*b[j];
                        c[i]%=k;
                    }
                for(i=0;i<len;i++)
                    b[i]=c[i];
            }
            REC tp;
            for(i=0;i<len;i++)
                for(j=0;j<len;j++)
                    for(e=0;e<len;e++){
                        tp.a[i][j]+=r.a[i][e]*r.a[e][j];
                        tp.a[i][j]%=k;
                    }
            r=tp;
            n/=2;
        }
        int ans=0;
        for(i=0;i<len;i++){
            ans+=b[i];
            ans%=k;
        }
        printf("%d\n",ans);
    }
    return 0;
}