求斐波那契数列前缀和（矩阵乘法）

最新推荐文章于 2023-11-25 13:29:42 发布

Sankkl1

最新推荐文章于 2023-11-25 13:29:42 发布

阅读量351

点赞数 1

分类专栏：知识点及模板整理文章标签：线性代数矩阵 c语言 c++

本文链接：https://blog.csdn.net/Sankkl1/article/details/119685260

版权

斐波那契数列模运算矩阵快速幂算法高效计算

关键词由CSDN通过智能技术生成

知识点及模板整理专栏收录该内容

49 篇文章 2 订阅

订阅专栏

设斐波那契数列 $f 1 = 1, f 2 = 1, f 3 = 2 . . .$ 求 $S_n \ mod\ m$ 的值

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

const int N = 3;

int n,m;

void mul(int c[], int a[], int b[][N])
{
	int tmp[N] = {0};
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			tmp[j] = (tmp[j] + (ll)a[i] * b[i][j]) % m;
	
	for(int i=0;i<N;i++) c[i] = tmp[i];
}

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]) % m;
	
	for(int i=0;i<N;i++)
		for(int j=0;j<N;j++)
			c[i][j] = tmp[i][j];
}

int main()
{
    int f[N] = {1, 1, 1};
    int a[N][N] = {
		{0, 1, 0},
		{1, 1, 1},
		{0, 0, 1}
	};
	scanf("%d%d",&n,&m);
	n--;
	while(n)
	{
		if(n & 1) mul(f, f, a);
		mul(a, a, a);
		n >>= 1;
	}
	printf("%d\n",f[2]);
    return 0;
}