矩阵快速幂模板


代码:

#pragma comment (linker,"/STACK:102400000,102400000")
#pragma comment (linker,"/STACK:102400000,102400000")
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#include <string>
#include <map>
#include <stack>
#include <vector>
#include <set>
#include <queue>

#define maxn 1005
#define MAXN 2005
#define INF 0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-6
#define MOD 1e9;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
#define FRE(i,a,b)  for(i = a; i <= b; i++)
#define FREE(i,a,b) for(i = a; i >= b; i--)
#define FRL(i,a,b)  for(i = a; i < b; i++)
#define FRLL(i,a,b) for(i = a; i > b; i--)
#define mem(t, v)   memset ((t) , v, sizeof(t))
#define sf(n)       scanf("%d", &n)
#define sff(a,b)    scanf("%d %d", &a, &b)
#define sfff(a,b,c) scanf("%d %d %d", &a, &b, &c)
#define pf          printf
#define DBG         pf("Hi\n")
const int MOD=1e9+9;
typedef long long LL;
using namespace std;
typedef vector<LL>vec;
typedef vector<vec>mat;
LL n;
mat mul(mat &A,mat &B)
{
    mat C(A.size(),vec(B[0].size()));
    for (int i=0; i<A.size(); i++)
    {
        for (int k=0; k<B.size(); k++)
        {
            for (int j=0; j<B[0].size(); j++)
                C[i][j]=(C[i][j]+A[i][k]*B[k][j])%MOD;
        }
    }
    return C;
}

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

void solve()
{
    mat A(2,vec(2));
    A[0][0]=1;
    A[0][1]=1;
    A[1][0]=1;
    A[1][1]=0;
    A=pow(A,n);
    pf("%lld\n",(A[1][0]%MOD));
}
int main()
{   while(~scanf("%lld",&n)){
        solve();
    }
    return 0;
}


你可能感兴趣的:(C++,c,矩阵快速幂)