- 加强版斐波那契 数学
链接:https://ac.nowcoder.com/acm/problem/15666
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
这是一个加强版的斐波那契数列。
给定递推式
求F(n)的值,由于这个值可能太大,请对109+7取模。
输入描述:
第一行是一个整数T(1 ≤ T ≤ 1000),表示样例的个数。
以后每个样例一行,是一个整数n(1 ≤ n ≤ 1018)。
输出描述:
每个样例输出一行,一个整数,表示F(n) mod 1000000007。
示例1
输入
复制
4
1
2
3
100
输出
复制
1
16
57
558616258
- 最主要的就是确定递推的矩阵
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
typedef struct node
{
ll v[6][6];
}mat;
mat mul(mat& A,mat& B)
{
mat ret;
for(int i=0;i<6;i++)
{
for(int j=0;j<6;j++)
{
ret.v[i][j]=0;
}
}
for(int i=0;i<6;i++)
{
for(int j=0;j<6;j++)
{
for(int k=0;k<6;k++)
{
ret.v[i][j]=(ret.v[i][j]+A.v[i][k]*B.v[k][j]%mod)%mod;
}
}
}
return ret;
}
mat pow(mat A,ll n)
{
mat z;
for(int i=0;i<6;i++)
{
for(int j=0;j<6;j++)
{
z.v[i][j]=0;
}z.v[i][i]=1;
}
while(n)
{
if(n&1) z=mul(A,z);
n>>=1;
A=mul(A,A);
}
return z;
}
int main()
{
int tt;
scanf("%d",&tt);
mat A;
for(int i=0;i<6;i++)
{
for(int j=0;j<6;j++)
{
A.v[i][j]=0;
}
}
A.v[0][0]=A.v[0][1]=A.v[0][2]=A.v[1][0]=A.v[2][2]=A.v[2][5]=A.v[3][3]=A.v[3][5]=A.v[4][4]=A.v[4][5]=A.v[5][5]=1;
A.v[3][4]=2;
A.v[2][3]=A.v[2][4]=3;
A.v[0][3]=A.v[0][5]=4;
A.v[0][4]=6;
while(tt--)
{
ll n;
scanf("%lld",&n);
if(n==0)
{
puts("0");
continue;
}
if(n==1)
{
puts("1");
continue;
}
mat ans=pow(A,n-1);
ll res=0;
for(int i=0;i<6;i++)
{
if(i==1) continue;
res+=ans.v[0][i];
res%=mod;
}
printf("%lld\n",res);
}
}
simplify ----------------------------------------------------
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
struct mat
{
ll v[6][6];
mat(){
memset(v,0,sizeof(v));
}
};
mat mul(mat& A,mat& B)
{
mat ret;
for(int i=0;i<6;i++)
{
for(int j=0;j<6;j++)
{
for(int k=0;k<6;k++)
{
ret.v[i][j]=(ret.v[i][j]+A.v[i][k]*B.v[k][j]%mod)%mod;
}
}
}
return ret;
}
mat pow(mat A,ll n)
{
mat z;
for(int i=0;i<6;i++)
{
z.v[i][i]=1;
}
while(n)
{
if(n&1) z=mul(A,z);
n>>=1;
A=mul(A,A);
}
return z;
}
int main()
{
int tt;
scanf("%d",&tt);
mat A;
A.v[0][0]=A.v[0][1]=A.v[0][2]=A.v[1][0]=A.v[2][2]=A.v[2][5]=A.v[3][3]=A.v[3][5]=A.v[4][4]=A.v[4][5]=A.v[5][5]=1;
A.v[3][4]=2;
A.v[2][3]=A.v[2][4]=3;
A.v[0][3]=A.v[0][5]=4;
A.v[0][4]=6;
while(tt--)
{
ll n;
scanf("%lld",&n);
if(n==0)
{
puts("0");
continue;
}
if(n==1)
{
puts("1");
continue;
}
mat ans=pow(A,n-1);
ll res=0;
for(int i=0;i<6;i++)
{
if(i==1) continue;
res+=ans.v[0][i];
res%=mod;
}
printf("%lld\n",res);
}
}