2018年湘潭大学程序设计竞赛 G-又见斐波那契 矩阵快速幂（矩阵构造）

题目链接:https://www.nowcoder.com/acm/contest/105/G
来源：牛客网
 

又见斐波那契

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld

题目描述

这是一个加强版的斐波那契数列。

给定递推式

求F(n)的值，由于这个值可能太大，请对109+7取模。

输入描述:

第一行是一个整数T(1 ≤ T ≤ 1000)，表示样例的个数。
以后每个样例一行，是一个整数n(1 ≤ n ≤ 1018)。

输出描述:

每个样例输出一行，一个整数，表示F(n) mod 1000000007。

 

示例1

输入

4
1
2
3
100

输出

1
16
57
558616258

 

代码如下：

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
struct matrix{
    ll aim[8][8];
    void clea(){
        memset(aim,0,sizeof(aim));
        for(int i=1;i<=6;i++)aim[i][i]=1;
    }
};
matrix mul(matrix a,matrix b){
    matrix ans;
    memset(ans.aim,0,sizeof(ans.aim));
    for(ll i=1;i<=6;i++){
        for(ll j=1;j<=6;j++){
            for(ll k=1;k<=6;k++){
                ans.aim[i][j]=(ans.aim[i][j]+a.aim[i][k]*b.aim[k][j])%mod;
            }
        }
    }
    return ans;
}
matrix quick(matrix a,ll k){
    matrix ans;
    ans.clea();
    while(k){
        if(k&1){
            ans=mul(ans,a);
        }
        a=mul(a,a);
        k/=2;
    }
    ans.aim[2][1]=0;ans.aim[3][1]=(ans.aim[3][1]*8)%mod;
    ans.aim[4][1]=ans.aim[4][1]*4%mod;ans.aim[5][1]=ans.aim[5][1]*2%mod;
    return ans;
}
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        ll n;
        scanf("%lld",&n);
        if(n==1){
            printf("1\n");
            continue;
        }
        if(n==2){
            printf("16\n");
            continue;
        }
        matrix a;
        memset(a.aim,0,sizeof(a.aim));
        for(int i=1;i<=6;i++)a.aim[i][1]=1;
        a.aim[1][2]=1;
        a.aim[3][3]=a.aim[6][3]=1;
        a.aim[4][3]=a.aim[5][3]=3;
        a.aim[4][4]=a.aim[6][4]=1;
        a.aim[5][4]=2;
        a.aim[5][5]=a.aim[6][6]=a.aim[6][5]=1;
        a=quick(a,n-1);
        ll ans=0;
        for(int i=1;i<=6;i++)
            ans=(ans+a.aim[i][1])%mod;
        printf("%lld\n",ans);
    }
    return 0;
}