【HDU1005】Number Sequence(矩阵快速幂)

记录一个菜逼的成长。。

题目链接
题目大意：
f[1] = 1,f[2] = 1,f[n] = (a*f[n-1]+b*f[n-2])%7(n > 2)
给你a,b。求f[n]。

网上好像都是找规律的。
我想用下矩阵求递推式，类似矩阵快速幂求斐波那契。
搞清楚原理后，只要稍微改动下就可以了。

我们可以构造矩阵
$\left(^{f[3]}_{f[2]}\right) = \left(^{a\space b}_{1\space 0}\right)\left(^{f[2]}_{f[1]}\right)$ $f[3] = a+b$
上式右边两个矩阵相乘是等于左边矩阵的。

不断递推会发现
$\left(^{f[4]}_{f[3]}\right) = \left(^{a^2+b\space ab}_{a\space b}\right)\left(^{f[2]}_{f[1]}\right)$ $f[4] = a^2+b+ab$
·
·
·
$\left(^{f[n]}_{f[n-1]}\right) = \left(^{a\space b}_{1\space 0}\right)^{n-2}\left(^{f[2]}_{f[1]}\right)$

设$A = \left(^{a\space b}_{1\space 0}\right)^{n-2}$
那么$ans = (A[0][0] + A[0][1]) \mod 7$
所以我们只要知道A就行了。

#include <bits/stdc++.h>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define fi first
#define se second
typedef long long LL;
typedef pair<int,int> PII;
const int INF = 0x3f3f3f3f;
const int MOD = 7;
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]);
                C[i][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(LL a,LL b)
{
    mat A(2,vec(2));
    A[0][0] = a;
    A[0][1] = b;
    A[1][0] = 1;
    A[1][1] = 0;
    A = pow(A,n-2);
    printf("%lld\n",(A[0][0]+A[0][1])%7);
}
int main()
{
    LL a,b;
    while(~scanf("%lld%lld%lld",&a,&b,&n),a)
        solve(a,b);
    return 0;
}