矩阵快速幂

Posted 2020-11-08 ww123

#include<bits/stdc++.h>
using namespace std;
#define Max 3
#define mod 10000
int m; //m阶矩阵
struct Matrix
{
    int m[Max][Max];
};
Matrix Mul(Matrix a,Matrix b)
{
    Matrix c;
    memset(c.m,0,sizeof(c.m));
    for(int i=0;i<m;i++)
    {
        for(int j=0;j<m;j++)
        {
            for(int k=0;k<m;k++)
            {
                c.m[i][j]+=((a.m[i][k]*b.m[k][j])%mod+mod)%mod;

            }
            c.m[i][j]=c.m[i][j]%mod;
        }
    }
    return c;
}
Matrix fastm(Matrix a,int n)
{
    Matrix res;
    res=a;
    memset(res.m,0,sizeof(res.m));
    for(int i=0;i<m;i++)
    res.m[i][i]=1;  //i==j位置设为1，初始化
    while(n)
    {
        if(n&1) res=Mul(res,a);
        n>>=1;
        a=Mul(a,a);
    }
    return res;
}
int main()
{
    int n;
    while(cin>>n,n!=-1)
    {
        Matrix a;
        m=2;
        //初始化矩阵
        a=fastm(a,n);
        cout<<a.m[1][0]<<endl;
    }
}