51Nod 1358 浮波那契 (矩阵快速幂)

1358 浮波那契

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

TengBieBie已经学习了很多关于斐波那切数列的性质，所以他感到一些些厌烦。现在他遇到了一个新的数列，这个数列叫做Float-Bonacci。这里有一个关于Float-Bonacci的定义。

对于一个具体的n，TengBieBie想要快速计算FB(n).

但是TengBieBie对FB的了解非常少，所以他向你求助。

你的任务是计算FB(n).FB(n)可能非常大，请输出FB(n)%1,000,000,007 (1e9+7)即可。

Input

输入共一行，在一行中给出一个整数n (1<=n<=1,000,000,000)。

Output

对于每一个n，在一行中输出FB(n)%1,000,000,007 (1e9+7)。

Input示例

5

Output示例

2

题意大家都明白，中文题，跟F(n) = F(n - 1) + F(n-2)斐波那契类似，唯一的坑爹的地方就是FB(n-3.4)

通过观察我们知道3.4*5=17,所以可以将FB(n)中的参数乘以5得到G(k) = G(k-5)+G(k-17);

而FB(n) = G(5n){为什么它们相等，原因:FB(n)=FB(n-1)+FB(n-3.4)->FB(5n)=FB(5n-5)+FB(5n-17),令k=5n,G(k)=G(k-5)+G(k-17)}

然后是构建进行运算的矩阵A了，要构建怎样的矩阵，必须知道初始化矩阵B是什么

初始化矩阵B = [G20 G19 G18 G17 ... ],总共17列，为什么呢，因为G(k) = G(k-5)+G(k-17),矩阵中必须有G(k-5),G(k-17),那么我们往上推一层就需要得到G(k+5),G(k),G(k-12),但是G(k-12)无法从已经有的矩阵中得到，所以这个矩阵就必须包括G(k-12),如此依次推导下去，需要G(k-7),G(k-2),(为什么变为k-14呢，因为k-2-5=k+3,而G(k+3)=G(k-2)+G(k-14),所以缺少一个G(k-14))G(k-14),G(k-9),G(k-4),G(k-16)....

而为什么初始化矩阵B是从G(20)开始呢，因为n<=4,5n<=20,所以我们必须从20开始进行构建初始化矩阵

如此正好是从k-1...k-17，共17列

接下来就可以写出矩阵A了

代码如下

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>
using namespace std;
const int mod = 1e9+7;
typedef long long LL;
typedef vector<LL>vec;
typedef vector<vec>mat;

mat mul(mat &A, mat &B) {
	mat C(A.size(), vec(B[0].size()));
	for(int i = 0; i < A.size(); i ++) {
		for(int k = 0; k < B.size(); k ++) {
			for(int j = 0; j < B[0].size(); j ++) {
				C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod;
			}
		}
	}
	return C;
}

mat pow(mat A, LL n) {
	mat B(A.size(), vec(A.size()));
	for(int i = 0; i < A.size(); i ++) B[i][i] = 1;
	while(n) {
		if(n & 1) B = mul(B, A);
		A = mul(A, A);
		n >>= 1;
	}
	return B;
}
LL n;
int main() {
	while(~scanf("%lld", &n)) {
		if(n <= 4) {
			printf("1\n");
			continue;
		}
		mat A(17, vec(17)), B(1,vec(17));
		A[0][0] = 1, A[12][0] = 1;//k-1->k+4
		A[1][1] = 1, A[13][1] = 1;//k-2->k+3
		A[2][2] = 1, A[14][2] = 1;//k-3->k+2
		A[3][3] = 1, A[15][3] = 1;//k-4->k+1
		A[4][4] = 1, A[16][4] = 1;//k-5->k
		//A[0][5] = 1;//k-6->k-1
		//A[1][6] = 1;//k-6->k-2
		//A[2][7] = 1;//k-7->k-3
		for(int i = 5; i < 17; i ++) {
			A[i- 5][i] = 1;
		}
		for(int i = 0; i < 17; i ++) {
			B[0][i] = 1;
		}
		A = pow(A, ((5LL * (n) - 20) + 4) / 5);//相差有多少个5,代表需要乘以多少次矩阵
		B = mul(B, A);
		printf("%lld\n", B[0][((5LL * n) - 20) % 5]);//得到相应的推导位置
	}
	return 0;
}