斐波那契前 n 项和 - 矩阵乘法快速幂

泠楠子

已于 2024-01-25 16:42:41 修改

阅读量98

点赞数

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

于 2023-10-08 19:34:48 首次发布

本文链接：https://blog.csdn.net/a1695574118/article/details/133688477

版权

模板同时被 2 个专栏收录

34 篇文章 0 订阅

订阅专栏

数论

7 篇文章 0 订阅

订阅专栏

本文介绍了一种使用矩阵快速幂的方法来计算斐波那契数列前n项和的C++代码实现，通过构造特定矩阵A并利用递归操作实现高效计算。

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

1303. 斐波那契前 n 项和 - AcWing题库

$F_{i}=[f_{i},f_{i+1},S_{i}]$

$F_{i+1}=[f_{i+1},f_{i+2},S_{i+1}]$

构造矩阵A使 $F_{i}*A=F_{i + 1}$

0 1 0

A = [ 1 1 1 ]

0 0 1

$F_{n}=F_{1}*A^{n-1}$

然后对这个式子进行快速幂，挺神奇的

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'
 
using namespace std;
 
typedef pair<int, int> PII;
typedef long long ll;
typedef long double ld;
 
const int N = 3;
 
int n, mod;
 
void mul(int C[], int A[], int B[][N])
{
	int tmp[N] = {0};
	//因为就一行所以没有i,相当于for(int i=0;i<1;i++)
	for(int j = 0; j < N; j ++)
		for(int k = 0; k < N; k ++)
			tmp[j] = (tmp[j] + (ll)A[k] * B[k][j]) % mod;
		
	memcpy(C, tmp, sizeof tmp);
}
 
void mul(int C[][N], int A[][N], int B[][N])
{
	int tmp[N][N] = {0};
	
	for(int i = 0; i < N; i ++)
		for(int j = 0; j < N; j ++)
			for(int k = 0; k < N; k ++)
				tmp[i][j] = (tmp[i][j] + (ll)A[i][k] * B[k][j]) % mod;
				
	memcpy(C, tmp, sizeof tmp);
}
 
int main()
{
	IOS
	int f1[N] = {1, 1, 1};
	int A[N][N] = {{0, 1, 0}, {1, 1, 1}, {0, 0, 1}};
	
	cin >> n >> mod;
	int k = n - 1;
	while(k)
	{
		if(k & 1)mul(f1, f1, A);//f1 = f1 * A 
		mul(A, A, A);//A = A * A
		k >>= 1;
	}
	
	cout << f1[2];
	
	
	return 0;
}