POJ 3070 矩阵mob

.

矩阵高速幂想法与快速幂相同

#include<iostream>
#include<cstdio>
#include<cstring>
#define MOD 10000
using namespace std;
struct matrix {
    int mat[][];
    matrix() {      //构造函数与结构体同名 若写作init()函数 需调用
        memset(mat,,sizeof(mat));
    }
};

matrix mul(matrix A , matrix B) {
    matrix C;
    for(int i=;i<=;i++) {
        for(int j=;j<=;j++) {
            for(int k=;k<=;k++) {
                C.mat[i][j]=(C.mat[i][j]+A.mat[i][k]*B.mat[k][j])%MOD;
            }
        }
    }
    return C;
}

matrix powmul(matrix A ,int n) {
    matrix B;  //初始化为单位阵
    B.mat[][]=;
    B.mat[][]=;
    while(n>=) {
        if(n&) {
            B=mul(B,A);
        }
        A=mul(A,A);    //自身幂次
        n/=;
    }
    return B;
}

int main()
{

    int n;
    while(~scanf("%d",&n))
    {
        if(n==-) break;
        matrix A;
        A.mat[][]=A.mat[][]=A.mat[][]=;
        A.mat[][]=;
        A=powmul(A,n);
        cout<<A.mat[][]<<endl;
    }

    return ;
}