原题链接: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();}