51nod1126(矩阵快速幂)

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1126

题意：中文题诶～

思路：构造矩阵：

　　

（ 0, 1 )^n-1    *    ( f0, f1 )

  ( b, a )                 ( f1, f2 )

代码：

#include <bits/stdc++.h>
#define mod 7
#define MAXN 2
using namespace std;

struct Matrix{
    int gg[MAXN][MAXN];
};

Matrix x;
Matrix temp={0, 1, 1, 1};
Matrix ans={1, 0, 0, 1};

Matrix multi(Matrix a, Matrix b){
    Matrix c;
    memset(c.gg, 0, sizeof(c.gg));
    for(int i=0; i<MAXN; i++){
        for(int j=0; j<MAXN; j++){
            for(int k=0; k<MAXN; k++){
                c.gg[i][j]+=a.gg[i][k]*b.gg[k][j]%mod;
            }
            c.gg[i][j]=(c.gg[i][j]+mod)%mod;
        }
    }
    return c;
}

int get_pow(int n){
    while(n){
        if(n&1){
            ans=multi(ans, x);
        }
        x=multi(x, x);
        n>>=1;
    }
    ans=multi(ans, temp);
    return (ans.gg[0][1]%mod+mod)%mod;
}

int main(void){
    int a, b, n;
    cin >> a >> b >> n;
    x.gg[0][0]=0, x.gg[0][1]=1, x.gg[1][0]=b, x.gg[1][1]=a;
    cout << get_pow(n-1) << endl;
}