2018年湘潭大学程序设计竞赛 矩阵快速幂

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

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

给定递推式

求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;
#define LL long long
#define N 100005
#define mod 1000000007
struct node
{
    LL a[7][7];
    node()
    {
        memset(a,0,sizeof(a));
    }
};

node operator *(node x,node y)
{
    node c;
    for(int i=1;i<=6;i++)
    {
        for(int j=1;j<=6;j++)
        {
            for(int k=1;k<=6;k++)
            {
                c.a[i][j]=(c.a[i][j]+((x.a[i][k]%mod*y.a[k][j]%mod)%mod))%mod;
            }
        }
    }
    return c;
}

void solve(LL k)
{
    node t1;
    t1.a[1][1]=1,t1.a[1][2]=0,t1.a[1][3]=8;
    t1.a[1][4]=4,t1.a[1][5]=2,t1.a[1][6]=1;
    node d;
    for(int i=1;i<=6;i++)d.a[i][1]=1;
    d.a[1][2]=1,d.a[3][3]=1,d.a[4][3]=3;
    d.a[5][3]=3;d.a[4][4]=1,d.a[5][4]=2;
    d.a[6][4]=1,d.a[5][5]=1,d.a[6][5]=1;
    d.a[6][6]=1;d.a[6][3]=1;
    while(k)
    {
        if(k&1)
            t1=t1*d;
        d=d*d;
        k>>=1;
        //printf("%lld %lld %lld %lld %lld %lld\n",t1.a[1][1],t1.a[1][2],t1.a[1][3],t1.a[1][4],t1.a[1][5],t1.a[1][6]);
    }
    printf("%lld\n",t1.a[1][1]);
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        LL n;
        scanf("%lld",&n);
        if(n==0)puts("0");
        if(n==1)puts("1");
        else solve(n-1);
    }
}