洛谷P1939（斐波那契数列）

题目链接： https://www.luogu.org/problemnew/show/P1939

题目描述

a[1]=a[2]=a[3]=1

a[x]=a[x-3]+a[x-1] (x>3)

求a数列的第n项对1000000007（10^9+7）取余的值。

输入输出格式

输入格式：

第一行一个整数T，表示询问个数。

以下T行，每行一个正整数n。

输出格式：

每行输出一个非负整数表示答案。

思路：直接循环一遍肯定TLE，用矩阵快速幂。

矩阵的构造： http://www.cnblogs.com/frog112111/archive/2013/05/19/3087648.html，这里说的很好。

上代码

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#define mod(x) ((x) % MOD)
using namespace std;
const int MOD = 1e9 + 7;
long long f[3];
struct mat
{
    long long m[10][10];
} unit;

mat a;
mat operator * (mat a, mat &b)
{
    mat ret;
    memset(ret.m, 0, sizeof(ret.m));

    for (int k = 0; k < 10; k++)
    {
        for (int i = 0; i < 10; i++)
        {
            if (a.m[i][k])
            {
                for (int j = 0; j < 10; j++)
                {
                    ret.m[i][j] = mod(ret.m[i][j] + a.m[i][k] * b.m[k][j]);
                }
            }
        }
    }
    return ret;
}

void init_unit()
{
    for (int i = 0; i < 10; i++)
    {
        unit.m[i][i] = 1;
    }
    return ;
}

mat pow_mat(mat a, long long n)
{
    mat ret = unit;
    while (n)
    {
        if (n & 1)
        {
            ret = ret * a;
        }
        n >>= 1;
        a = a * a;
    }
    return ret;
}

void ANS_Cf(mat a)
{ 
     f[0] = (a.m[0][0] %MOD+a.m[1][0] %MOD +a.m[2][0] %MOD) %MOD;
     f[1] = (a.m[0][1]  %MOD+a.m[1][1] %MOD +a.m[2][1] %MOD) %MOD;
     f[2] = (a.m[0][2] %MOD +a.m[1][2] %MOD +a.m[2][2] %MOD )%MOD;
    return ;
}


int main()
{
    long long n,t;
    scanf("%lld",&t);
	while (t--)
    {
        scanf("%lld",&n);
		if (n == 1||n==2||n==3)
        {
            cout << '1' << '\n';
            continue;
        }
        init_unit();
        a.m[1][0] = a.m[0][2] = a.m[2][1] = a.m[2][2] =1;
        a.m[1][1] = a.m[0][1] =  a.m[1][2] = a.m[0][0] = a.m[2][0] = 0;
        a = pow_mat(a, n - 1);
        ANS_Cf(a);
        cout << f[0] << '\n';
    }
    return 0;
}