首页 > 代码库 > 矩阵乘法

矩阵乘法

#include<bits/stdc++.h>
using namespace std;
const int N=2;
const int MOD=10000;
struct MAT{int a[N][N];};

MAT operator*(MAT x, MAT y){
    MAT z;
    for(int i=0;i<N;++i)
        for(int j=0;j<N;++j){
            z.a[i][j]=0;
            for(int k=0;k<N;++k)
                z.a[i][j]=(z.a[i][j]+x.a[i][k]*y.a[k][j])%MOD;
        }
    return z;
}

MAT pow(int n){
    MAT ret,s;
    ret.a[0][0]=ret.a[1][1]=1;
    ret.a[1][0]=ret.a[0][1]=0;
    s.a[0][0]=s.a[0][1]=s.a[1][0]=1;
    s.a[1][1]=0;
    for(;n;n>>=1,s=s*s)if(n&1)ret=ret*s;
    return ret;
}

int main(){
    int n;
    while(cin>>n,n>=0){
        MAT ans=pow(n);
        cout<<ans.a[0][1]<<endl;
    }
}

矩阵乘法