题目描述
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;
由于 f[n]=f[n-1]+f[n-3] 构造矩阵时要满足这个式子, 所以有{ f[n] ,f[n-1], f[n-2] }={ f[n-1],f[n-2],f[n-3] } X 矩阵A (不是差乘)
f[n]=1*f[n-1]+0*f[n-2]+1*f[n-3] 所以第1列为 1 0 1 是列
同理由于已经满足了 f[n]的关系式所以不去满足剩下的f[n-1] f[n-2] 有
f[n-1] = 1*f[n-1] + 0*f[n-2]+ 0*f[n-3] 第2列为1 0 0
f[n-2]=0*f[n-1] + 1*f[n-2] + 0*f[n-3] 第3列为0 1 0
A = 1 1 0
0 0 1
1 0 0
#include <math.h>
#include <algorithm>
#include <iostream>
#include <queue>
#include <cstdio>
#include <list>
#include <deque>
#include <set>
#include <vector>
#include <map>
#include <cstring>
#include <iomanip>
#define max(a,b) (a>b?a:b)
#define LL long long
using namespace std;
const int mod=1e9+7;
struct node
{
LL a[120][120];
};
int n;
node getnum(node x,node y)
{
node z;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
z.a[i][j]=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
{
z.a[i][j]=(z.a[i][j]%mod+x.a[i][k]*y.a[k][j]%mod)%mod;
}
return z;
}
node Pow(node a,LL b)
{
/* for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
z.a[i][j]=0;
}*/
node ans;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
if(i==j)ans.a[i][j]=1;
else ans.a[i][j]=0;
}
while(b!=0)
{
if(b&1!=0)
ans=getnum(ans,a);
a=getnum(a,a);
// getnum(a,a,z);
//getnum(a,a,a);
b>>=1;
}
return ans;
}
//int num[1000];
node num;
int main()
{
ios::sync_with_stdio(false);
LL n1;
int t;
cin>>t;
while(t--)
{
cin>>n1;
int a=1,b=1,c=1;
n=3;
for(int i=0;i<3;i++)
for( int j=0;j<3;j++)
{
num.a[i][j]=0;
if((i==0&&j==0)||(i==0&&j==1)||(i==1&&j==2)||(i==2&&j==0))
num.a[i][j]=1;
}
node ans=Pow(num,n1);
/* for(int i=0;i<3;i++)
for(int j=0;j<3;j++)
{
if(j==2)cout<<ans.a[i][j]<<endl;
else cout<<ans.a[i][j]<<" ";
}*/
cout<<ans.a[0][1]<<endl;
}
return 0;
}
结果为什么是a[0][1] 由于算的是n+1的 当然输出矩阵也可以发现