Luogu1939 【模板】矩阵加速（数列）

原题链接：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。

输出格式：

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

输入输出样例

输入样例#1：

3
6
8
10

输出样例#1：

4
9
19

说明

对于30%的数据 n<=100；

对于60%的数据 n<=2*10^7；

对于100%的数据 T<=100，n<=2*10^9；

题解

不知道为什么递归快速幂出现了玄妙的错误，向非递归低头。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int M=105,mod=1e9+7;
ll n=3,T;
struct matrix{ll sq[M][M];matrix(){memset(sq,0,sizeof(sq));}}mat,ans,one;
matrix operator *(matrix a,matrix b)
{
    matrix c;
    for(int i=1;i<=n;++i)for(int j=1;j<=n;++j)for(int k=1;k<=n;++k)
    (c.sq[i][j]+=a.sq[i][k]*b.sq[k][j]%mod)%=mod;
    return c;
}
void out(matrix a){for(int i=1;i<=3;++i,putchar(10))for(int j=1;j<=3;++j)printf("%d ",a.sq[i][j]);}
matrix power(matrix x,ll p)
{
    matrix ans=one;
    for(;p;p>>=1,x=x*x)if(p&1)ans=ans*x;
    return ans;
}
void in(){scanf("%d",&T);}
void ac()
{
    int a;
    one.sq[1][1]=one.sq[2][2]=one.sq[3][3]=1;
    mat.sq[1][1]=mat.sq[3][1]=mat.sq[1][2]=mat.sq[2][3]=1;
    while(T--)scanf("%d",&a),printf("%lld\n",power(mat,a-1).sq[1][1]);
}
int main(){in();ac();}