斐波那契数列(III)(矩阵快速幂)

斐波那契数列(III)

题目

在这里插入图片描述

答案

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;

ull Pow(ull n)
{
	ull a[2][2]={{1,1},{1,0}},tmp[2][2]={{1,0},{0,1}},tmp1,tmp2,tmp3,tmp4;
	while(n)
	{
		if(n&1)
		{
			tmp1=tmp[0][0]*a[0][0]+tmp[0][1]*a[1][0];
			tmp2=tmp[0][0]*a[0][1]+tmp[0][1]*a[1][1];
			tmp3=tmp[1][0]*a[0][0]+tmp[1][1]*a[1][0];
			tmp4=tmp[1][0]*a[0][1]+tmp[1][1]*a[1][1];
			tmp[0][0]=tmp1%998244353,tmp[0][1]=tmp2%998244353,tmp[1][0]=tmp3%998244353,tmp[1][1]=tmp4%998244353;
		}
		tmp1=a[0][0]*a[0][0]+a[0][1]*a[1][0];
		tmp2=a[0][0]*a[0][1]+a[0][1]*a[1][1];
		tmp3=a[1][0]*a[0][0]+a[1][1]*a[1][0];
		tmp4=a[1][0]*a[0][1]+a[1][1]*a[1][1];
		a[0][0]=tmp1%998244353,a[0][1]=tmp2%998244353,a[1][0]=tmp3%998244353,a[1][1]=tmp4%998244353;
		n>>=1;
	}
	return tmp[0][0]%998244353;
}

int main()
{
	ull n;
	cin>>n;
	cout<<Pow(n-1);
}

注意

  1. 要先将每次更新后的矩阵值保存在tmp1到tmp4中,再在最后赋值(如果按顺序依次赋值,那么前面的值的改变会影响后面的运算,进而导致结果错误)
  2. 本题将记得取模,并将所有变量设置为`unsigned long long

总结

这道题采用了矩阵快速幂,详细方法可到下方链接中查找
https://zhuanlan.zhihu.com/p/54378423

同时,我采用的快速幂和原方法不同,我的方法类似于我的这篇文章——还在用pow函数做幂运算相关的算法题?快速幂表示不服!

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
斐波数列矩阵快速实现方式如下所示: 首先,我们定义一个2阶矩阵: \[ M = \begin{bmatrix} 1\enspace 1\\ 1\enspace 0 \end{bmatrix} \] 然后,根据斐波数列的递推公式\[ F_n = F_{n-1} + F_{n-2} \],我们可以将这个公式抽象成矩阵运算的形式: \[ \begin{bmatrix} F_{n} \\ F_{n-1} \end{bmatrix} = M^{n-1} \cdot \begin{bmatrix} F_1 \\ F_0 \end{bmatrix} \] 其中,\[ F_1 \]和\[ F_0 \]分别表示斐波数列的初始值。 通过快速运算的思想,我们可以将指数为正整数的运算复杂度从\[ O(n) \]降低到\[ O(\log_2 n) \],而且矩阵乘法运算也符合结合律,因此可以使用矩阵快速来计算斐波数列。 代码实现如下: ```python import numpy as np def fibonacci_matrix_power(n, f1, f0): M = np.array([[1, 1], [1, 0]]) # 定义2阶矩阵M result = np.array([[f1], [f0]]) # 初始值矩阵 power = np.eye(2, dtype=int) # 运算的初始值为单位矩阵 while n > 0: if n % 2 == 1: power = np.dot(power, M) # 如果n是奇数,将M乘到结果矩阵上 M = np.dot(M, M) # 将M自乘,n每次右移一位 n //= 2 result = np.dot(power, result) # 最终的结果矩阵 return result # 返回斐波数列的第n项 # 测试示例 n = 10 f1 = 1 f0 = 0 fib_n = fibonacci_matrix_power(n, f1, f0) print(fib_n) ``` 这段代码通过矩阵快速的方式计算斐波数列的第n项,其中参数n为要计算的项数,f1和f0分别为斐波数列的初始值。运行结果为斐波数列的第n项的值。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值