CSP-S数论从入门到入土- 矩阵快速幂

1.基本概念
        1.1原理

        矩阵快速幂,即为用快速幂加速矩阵乘法操作,使得矩阵高次幂的计算时间复杂度从O(N)变成O(\log N),大大减少用时

        为什么可以对矩阵进行快速幂优化呢?这是因为矩阵乘法满足结合律

        \because C[i][j]=\sum _{k=1}^{m} A[i][k] * B [i][k]

        所以在做两次操作后 

        \therefore C_{2}[i][j] = \sum _{k=1}^{m}\textrm{A2[i][k]*B[i][k]}

        所以矩阵乘法满足结合律

        利用快速幂思想,我们不难写出代码

        1.2实现

        

#include<bits/stdc++.h>
using namespace std;
#define int long long
int a,b,c,d,m,n,k,z[2][2],z2[2][2];
void zc(int a[2][2],int b[2][2]){
	int c[2][2];
	for(int i=0;i<2;i++) for(int j=0;j<2;j++) c[i][j]=0;
	for(int i=0;i<2;i++)//矩阵乘
		for(int j=0;j<2;j++)
			for(int k=0;k<2;k++)
				c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j]%m)%m; //矩阵公式
	for(int i=0;i<2;i++) for(int j=0;j<2;j++) a[i][j]=c[i][j];
}
signed main(){
	c=1,d=1,a=1,b=1,m=1e9+7;
	z[0][0]=1;
	z[0][1]=1;
	z[1][0]=1;
	z[1][1]=1;
	z2[0][0]=1;
	z2[0][1]=0;
	z2[1][0]=0;
	z2[1][1]=1;
	int zzz=13170720;
	while(zzz){//快速幂
		if(zzz&1){
			zc(z2,z);
		}
		zc(z,z);
		zzz>>=1;
	}
	cout<<z[0][0]<<endl;
}
      2.矩阵快速幂扩展运用

        我们了解了用快速幂加速矩阵乘法的算法,但是,在比赛中,很少出现很明显指出与矩阵相关的题目,更多时候,是让你用快速幂去加速一个递推式。

        碰到这种题目,第一反应是根据我们对于高中数学中数列的理解,转化递推式。但是,在转换后,我们往往会得出一个含有无理数的式子,这种式子就不好用快速幂处理了

        我们不妨以用O(\log N)时间复杂度算法求出斐波那契数列第N项这一问题为例

        斐波那契递推式是A_{n}=A_{n-1}+A_{n-2}

        不难发现 该式子等价于 \begin{bmatrix} A_{n}\\ A_{n+1} \end{bmatrix} * \begin{bmatrix} 0 & 1 \\ 1 & 1 \end{bmatrix} = \begin{bmatrix} A_{n+1}\\ A_{n+2} \end{bmatrix}

         我们不妨将其转换成一个2*1的矩阵乘上一个2*2的矩阵的矩阵乘法问题

        给出代码

        

#include<bits/stdc++.h>
using namespace std;
#define int long long
int a,b,c,d,m,n,z[2][2],z2[2][2];
void zc(int a[2][2],int b[2][2]){
	int c[2][2];
	for(int i=0;i<2;i++) for(int j=0;j<2;j++) c[i][j]=0;
	for(int i=0;i<2;i++)
		for(int j=0;j<2;j++)
			for(int k=0;k<2;k++)
				c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j]%m)%m;
	for(int i=0;i<2;i++) for(int j=0;j<2;j++) a[i][j]=c[i][j];
}
signed main(){
	cin>>c>>d>>a>>b>>n>>m;
	z[0][0]=0;
	z[0][1]=1;
	z[1][0]=1;
	z[1][1]=1;
	z2[0][0]=1;
	z2[0][1]=0;
	z2[1][0]=0;
	z2[1][1]=1;
	int zzz=n-1;
	while(zzz){
		if(zzz&1){
			zc(z2,z);
		}
		zc(z,z);
		zzz>>=1;
	}
	cout<<(1ll*a*z2[0][0]+1ll*b*z2[0][1])%m<<endl;
}

  • 14
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值