poj 3070 矩阵快速幂模板

题意：求fibonacci数列第n项

 

#include "iostream"
#include "vector"
#include "cstring"
using namespace std;

typedef unsigned long int ULL;
typedef vector<ULL> vec;
typedef vector<vec> mat;
const ULL P=10000;
int n,m;

mat mul(mat &A,mat &B)      //return A*B
{
    mat C(A.size(),vec(B[0].size()));
    for (int i=0;i<(int)A.size();i++)
    {
        for (int k=0;k<(int)B.size();k++)
        {
            for (int j=0;j<(int)B[0].size();j++)
            {
                C[i][j]=(C[i][j]+A[i][k]*B[k][j])%P;
            }
        }
    }
    return C;
}

mat m_pow(mat A,int m)      //return A^m
{
    mat B(A.size(),vec(A.size()));
    for (int i=0;i<(int)A.size();i++)
        B[i][i]=1;
    while (m>0)
    {
        if (m&1)    B=mul(B,A);
        A=mul(A,A);
        m>>=1;
    }
    return B;
}

int main()
{
    while (cin>>n)
    {
        if (n==-1)  break;
        mat A(2,vec(2));

        A[0][0]=1;  A[0][1]=1;
        A[1][0]=1;  A[1][1]=0;

        A=m_pow(A,n);
        cout<<A[1][0]<<endl;
    }
}