一、题目链接
二、题目大意
求斐波拉契数列的第
n
n
n项,并对
1
0
9
+
7
10^9+7
109+7取余。
数据范围:
1
≤
n
<
2
63
1 \leq n < 2^{63}
1≤n<263。
三、题目分析
-
很容易知道在 2 63 2^{63} 263的数据范围下无法像传统方式那样,在 O ( n ) O(n) O(n)时间递推完成这个过程。而且因为取余运算和精度问题,我们也不能用斐波拉契数列的通项公式来做。所以必须另寻他法。
-
我们很容易知道,我们设置前后两个状态: A n − 1 A_{n-1} An−1和 A n A_n An, A n A_n An和 A n − 1 + A n A_{n-1}+A_n An−1+An,所以每一次变化的方式都是恒定的,我们可以用矩阵快速幂来解决这个问题。
-
构建如下矩阵:
{ A n A n − 1 A n − 1 A n − 2 } (1) \begin{Bmatrix} An & An-1 \\ An-1 & An-2 \end{Bmatrix} \tag{1} {AnAn−1An−1An−2}(1){ 1 1 1 0 } (2) \begin{Bmatrix} 1 & 1 \\ 1 & 0 \end{Bmatrix} \tag{2} {1110}(2)
上面两个相乘,且根据斐波拉契数列,会得到这样一个结果:
{ A n + 1 A n A n A n − 1 } (3) \begin{Bmatrix} An+1 & An \\ An & An-1 \end{Bmatrix}\tag{3} {An+1AnAnAn−1}(3)此时我们就能够不断得到斐波拉契数列后续的项了,而因为每次所乘的矩阵都相同,所以我们可以使用矩阵快速幂来实现这一过程。
4.因为矩阵的行列不大,所以可以当作常数处理,最终,时间复杂度就是快速幂得时间复杂度,为 O ( l o g n ) O(logn) O(logn)。
四、正解代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define mod 1000000007
using namespace std;
typedef long long ll;
struct node
{
ll a[5][5];
};
ll n,K;
node init()
{
node res;
for(ll i=1;i<=2;i++)
{
for(ll j=1;j<=2;j++)
{
if(i==j)
res.a[i][j]=1;
else
res.a[i][j]=0;
}
}
return res;
}
node matrix_mul(node temp1,node temp2)
{
node ans;
for(ll i=1;i<=2;i++)
{
for(ll j=1;j<=2;j++)
{
ans.a[i][j]=0;
for(ll k=1;k<=2;k++)
ans.a[i][j]=(ans.a[i][j]+temp1.a[i][k]*temp2.a[k][j]%mod)%mod;
}
}
return ans;
}
node matrix_pow(node temp,ll p)
{
node ans=init();
while(p)
{
if(p&1)
ans=matrix_mul(ans,temp);
p>>=1;
temp=matrix_mul(temp,temp);
}
return ans;
}
int main()
{
scanf("%lld",&K);
node A;
A.a[1][1]=A.a[1][2]=A.a[2][1]=1;
A.a[2][2]=0;
if(K==1)
{
printf("1");
return 0;
}
A=matrix_pow(A,K-1);
printf("%lld",A.a[1][1]);
return 0;
}