模拟赛 数列

题目描述

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;
}