矩阵快速幂 -codevs斐波拉契数列2

最新推荐文章于 2022-01-22 13:02:11 发布

silentsaber~

最新推荐文章于 2022-01-22 13:02:11 发布

阅读量357

点赞数 1

分类专栏：数学矩阵乘法文章标签：数学矩阵乘法

本文链接：https://blog.csdn.net/nikelong0/article/details/50748537

版权

数学同时被 2 个专栏收录

14 篇文章 0 订阅

订阅专栏

矩阵乘法

9 篇文章 0 订阅

订阅专栏

1732 Fibonacci数列 2

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 钻石 Diamond

在“1250 Fibonacci数列”中，我们求出了第n个Fibonacci数列的值。但是1250中，n<=10 9 。现在，你的任务仍然是求出第n个Fibonacci数列的值，但是注意：n为整数，且1 <= n <= 100000000000000

输入描述 Input Description

输入有多组数据，每组数据占一行，为一个整数n(1 <= n <= 100000000000000)

输出描述 Output Description

输出若干行。每行输出第（对应的输入的）n个Fibonacci数（考虑到数会很大，mod 1000000007）

样例输入 Sample Input

3
4
5

样例输出 Sample Output

2
3
5

数据范围及提示 Data Size & Hint

1 <= n <= 100000000000000

数据大了会超时？开玩笑，2^3 log n这么小的数量级，秒秒钟跑完，重载运算符跑一下就好了，思路和斐波拉契1一样。

构造一个T矩阵，自乘n-2次，再和初始矩阵相乘，取mat[1][1]就好了。

注意因为数据大了，有可能乘起来爆int ，因为两个数都很接近mod的话，乘起来是爆int的，改用long long

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#define LL long long
using namespace std;
const LL maxn=5;
LL n;
const LL M=1000000007;
struct Mat
{
	LL mat[maxn][maxn];
};
Mat operator*(Mat a,Mat b)
{
	Mat c;
	memset(c.mat,0,sizeof(c.mat));
	for(LL k=1;k<=2;k++)
	{
		for(LL i=1;i<=2;i++)
		{
			for(LL j=1;j<=2;j++)
			{
				c.mat[i][j]+=(a.mat[i][k]%M*b.mat[k][j]%M);
				c.mat[i][j]%=M;
			}
		}
	}
	return c;
}
Mat operator ^(Mat a,LL k)
{
	Mat c;
	memset(c.mat,0,sizeof(c.mat));
	for(LL i=1;i<=2;i++)
	{
		for(LL j=1;j<=2;j++)c.mat[i][j]=(i==j);
	}
	for(;k;k>>=1)
	{
		if(k&1)c=c*a;
		a=a*a;
	}
	return c;
}
int main()
{
	freopen("2.out","w",stdout);
	while(scanf("%lld",&n)!=EOF)
	{
		if(n<3)printf("1\n");
		else
		{
			Mat f,g;
			memset(f.mat,0,sizeof(f.mat));
			f.mat[1][1]=f.mat[1][2]=f.mat[2][1]=1;
			g=f^(n-2);
			f.mat[1][2]=0;
			g=g*f;
			printf("%lld\n",g.mat[1][1]);
		}		
	}
	return 0;
}