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

最新推荐文章于 2021-08-17 22:48:21 发布

森屿～

最新推荐文章于 2021-08-17 22:48:21 发布

阅读量152

点赞数

分类专栏：数论文章标签：矩阵快速幂斐波那契数列大整数模运算算法 C++

本文链接：https://blog.csdn.net/qq_46667619/article/details/115490685

版权

数论专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目链接

题目描述

这是一个加强版的斐波那契数列。
给定递推式在这里插入图片描述
求F(n)的值，由于这个值可能太大，请对109+7取模。

输入描述：

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

输出描述：

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

思路:

由数据范围知道是矩阵快速幂求解，推出
初始矩阵为
A= $\begin{gathered} \begin{bmatrix} 1 & 1 &1 &1 &1 &1 \\ 1 & 0 &0&0&0&0 \\ 0 & 0 &1&3&3&1 \\ 0 & 0 &0&1&2&1 \\ 0 & 0 &0&0&1&1 \\ 0 & 0 &0&0&0&1 \end{bmatrix} \quad \end{gathered}$
目标矩阵为：
B(i)= $\begin{gathered} \begin{bmatrix} F(i) \\ F(i-1) \\ (i+1)^3 \\(i+1)^2 \\ i+1 \\ 1 \end{bmatrix} \quad \end{gathered}$
所以B(i+1)=A*B(i),F(n)为ans矩阵乘以A的n-1次方后矩阵的第一个元素，n为1、0时需特判。解题也可以直接使用数组（应该可以减少代码量）。

code：

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#define debug(x) cout << #x << ":" << x << endl;
#define fast ios::sync_with_stdio(false), cin.tie(0);
#define fi first
#define se second
using namespace std;
typedef long long ll;
const double eps = 1e-10;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1.0);
const int mod = 1e9 + 7;
const int maxn = 2e6 + 10;

ll n;

struct mat{
    ll m[6][6];
    /*
    可添加构造函数
    mat(){
    	memset(m,0,sizeof m);
    }
    */
} base, ans;

void init()
{
    memset(base.m,0,sizeof base.m);
    base.m[0][0] = base.m[0][1] = base.m[0][2] = base.m[0][3] = base.m[0][4] = base.m[0][5] = 1;
    base.m[2][2] = 1, base.m[2][3] = base.m[2][4] = 3,base.m[2][5] = 1;base.m[1][0] = 1;
    base.m[3][3] = base.m[3][5] = 1, base.m[3][4] = 2;
    base.m[4][4] = base.m[4][5] = 1;
    base.m[5][5] = 1;ans.m[0][0] = 1;
    ans.m[2][0] = 8, ans.m[3][0] = 4, ans.m[4][0] = 2, ans.m[5][0] = 1;
}

mat mul(mat a,mat b)
{
    mat ret;
    for (int i = 0; i < 6;i++)
        for (int j = 0; j < 6;j++)
            ret.m[i][j] = 0;
    for (int i = 0; i < 6; i++){
        for (int j = 0; j < 6; j++){
             for (int k = 0; k < 6; k++){
                        ret.m[i][j] = (ret.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
             }
        }
    }
    return ret;
}

void ppow(mat a,ll b)
{
    while(b){
        if(b&1)
            ans = mul(a, ans);
        b >>= 1;
        a = mul(a, a);
    }
}

int main()
{
    fast;
    int t;
    cin >> t;
    while(t--){
        cin >> n;
        init();
        if(n<=1){
            if(n==1)
                cout << 1 << endl;
            else
                cout << 0 << endl;
            continue;
        }
        ppow(base,n-1);
        cout << ans.m[0][0] << endl;
        // cout << ans.m[1][0] << endl;
    }
    return 0;
}