Jzzhu and Sequences CodeForces - 450B (矩阵快速幂)

传送门

题解：推导出公式，使用矩阵快速幂解决 

附上代码:

 
#include<bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
 
const ll mod=1e9+7;
const int N=2;
 
struct node{
    ll a[3][3];
};
node shu,ans,mp;
 
node matrix(node x,node y)
{
    for(int i=1;i<=N;i++){
        for(int j=1;j<=N;j++){
            mp.a[i][j]=0;
            for(int p=1;p<=N;p++){
                mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]+mod)%mod;
            }
        }
    }
    return mp;
}
 
void work(ll k)
{
    for(int i=1;i<=N;i++){
        for(int j=1;j<=N;j++){
            ans.a[i][j]=0;
        }
    }
    for(int i=1;i<=N;i++){
        ans.a[i][i]=1;
    }
    node t=shu;
    while(k){
        if(k&1){
            ans=matrix(ans,t);
        }
        k>>=1;
        t=matrix(t,t);
    }
}
 
int main()
{
    ll x,y,n;
    scanf("%lld%lld",&x,&y);
    scanf("%lld",&n);
    if(n==1){
        printf("%lld\n",(x%mod+mod)%mod);
    }else if(n==2){
        printf("%lld\n",(y%mod+mod)%mod);
    }else{
        shu.a[1][1]=1;
        shu.a[1][2]=-1;
        shu.a[2][1]=1;
        shu.a[2][2]=0;
        work(n-2);
        printf("%lld\n",(((ans.a[1][1]*y+ans.a[1][2]*x)%mod)+mod)%mod);
    }
    return 0;
}