斐波那契数列-矩阵快速幂

最新推荐文章于 2022-10-20 14:59:15 发布

窦尊

最新推荐文章于 2022-10-20 14:59:15 发布

阅读量100

点赞数

分类专栏：算法文章标签：矩阵算法线性代数

本文链接：https://blog.csdn.net/m0_56511205/article/details/126129445

版权

算法专栏收录该内容

21 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用快速幂和矩阵乘法优化斐波那契数列(Fibonacci sequence)的计算，避免了递归和高时间复杂度的问题。通过初始化单位矩阵和系数矩阵，结合矩阵乘法，实现了在大数情况下快速求解Fn的高效算法。代码示例展示了C++实现的过程。

摘要由CSDN通过智能技术生成

Fn=1 (n≤2)，Fn = Fn−1+Fn−2 (n≥3)，求给出一个n，求Fn？

分析：显然当n非常大时，使用递推的方法复杂度太高。此时需要考虑快速幂提高速度，以及矩阵乘法进行优化

ans矩阵用来存放答案，首先应给ans初始化成单位矩阵，即主对角线的值赋值成1。

base矩阵用来存放幂，但应初始化为系数矩阵，本例中为 1 1

1 0

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdlib.h>
#define ll long long

const int mod = 1e9+7;

using namespace std;
struct Node
{
	ll s[10][10];
}ans,base;

Node mul(Node x,Node y)
{
	Node temp;
	memset(temp.s,0,sizeof(temp.s));
	
	for(int i=1;i<=2;i++)
	{
		for(int j=1;j<=2;j++)
		{
			for(int k=1;k<=2;k++)
			{
				temp.s[i][j] +=  (x.s[i][k]%mod)*(y.s[k][j]%mod);
				temp.s[i][j] %= mod;
			}
		}
	}
	
	
	return temp;		
}

Node ksm(ll n)
{
	while(n)
	{
		if(n&1) ans = mul(ans,base);
		base = mul(base,base);
		n>>=1;    
	}
}


int main()
{
	memset(ans.s,0,sizeof(ans.s));
	memset(base.s,0,sizeof(base.s));
	
	base.s[1][1] = base.s[1][2] =base.s[2][1] = 1;
	ans.s[1][1] = ans.s[2][2] = 1;
	ll n;
	cin>>n;
	
	if(n<3)
	{
		cout<<1;
		return 0;
	}
	
	ksm(n);
	
	cout<<ans.s[2][1];
	
	
	return 0;
}