hdu 1757 矩阵快速幂 **

一看正确率这么高，以为是水题可以爽一发，结果是没怎么用过的矩阵快速幂，233

题解链接：点我

#include<iostream>
#include<cstring>
const int N=10;
using namespace std;
int k,m;
struct Matrix{
    int map[N][N];
};

Matrix matrix;

void Initiate(){
    for(int i=0;i<N;i++){
        scanf("%d",&matrix.map[0][i]);
    }
    for(int i=1;i<N;i++){
        for(int j=0;j<N;j++){
            if(i==(j+1))matrix.map[i][j]=1;
            else matrix.map[i][j]=0;
        }
    }
}

//矩阵相乘
Matrix Mul(Matrix &a,Matrix &b){
    Matrix c;
    for(int i=0;i<N;i++){
        for(int j=0;j<N;j++){
            c.map[i][j]=0;
            for(int k=0;k<N;k++){
                c.map[i][j]+=a.map[i][k]*b.map[k][j];
            }
            c.map[i][j]%=m;
        }
    }
    return c;
}

//快速幂
Matrix Pow(int n){
    Matrix t;
    if(n==1)return matrix;
    if(n&1)return Mul(matrix,Pow(n-1));
    else {
        Matrix temp=Pow(n>>1);
        return Mul(temp,temp);
    }
}


int main(){
    while(scanf("%d%d",&k,&m)!=EOF){
        Initiate();
        if(k<10){
            printf("%d
",k%m);
            continue;
        }
        Matrix temp=Pow(k-9);
        int ans=0;
        for(int i=0;i<N;i++){
            ans+=temp.map[0][i]*(N-i-1); //最后要乘上f[9],f[8],...,f[1],f[0];
            ans%=m;
        }
        printf("%d
",ans);
    }
    return 0;
}