题目描述
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。
输出格式
每行输出一个非负整数表示答案。
样例输入
3
6
8
10
样例输出
4
9
19
数据范围
对于30%的数据 n<=100;
对于60%的数据 n<=2*10^7;
对于100%的数据 T<=100,n<=2*10^9;
题解
矩阵乘法+快速幂
如果想要数据可以在评论中叫我
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#define mod 1000000007
#define ll long long
using namespace std;
int t,n;
ll a[3][3],b[3][3];
void clean()
{
memset(a,0,sizeof(a)); memset(b,0,sizeof(b));
for(int i=0;i<3;i++) b[i][i]=1;
a[0][2]=a[1][0]=a[2][1]=a[2][2]=1;
}
void mul(ll a[3][3],ll b[3][3],ll ans[3][3])
{
ll t[3][3];
for(int i=0;i<3;i++)
for(int j=0;j<3;j++)
{t[i][j]=0;
for(int k=0;k<3;k++)
t[i][j]=(t[i][j]+a[i][k]*b[k][j])%mod;
}
for(int i=0;i<3;i++)
for(int j=0;j<3;j++)
ans[i][j]=t[i][j];
}
void ksm(int k)
{
while(k>0)
{if(k&1) mul(a,b,b);
mul(a,a,a);
k=k>>1;
}
}
int main()
{
//freopen("seq.in","r",stdin);
//freopen("seq.out","w",stdout);
scanf("%d",&t);
while(t--)
{clean();
scanf("%d",&n);
ksm(n-1);
ll ans=0;
for(int i=0;i<3;i++)
ans=(ans+b[i][0])%mod;
printf("%d\n",ans);
}
return 0;
}