1.基本概念
1.1原理
矩阵快速幂,即为用快速幂加速矩阵乘法操作,使得矩阵高次幂的计算时间复杂度从变成
,大大减少用时
为什么可以对矩阵进行快速幂优化呢?这是因为矩阵乘法满足结合律
所以在做两次操作后
所以矩阵乘法满足结合律
利用快速幂思想,我们不难写出代码
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.矩阵快速幂扩展运用
我们了解了用快速幂加速矩阵乘法的算法,但是,在比赛中,很少出现很明显指出与矩阵相关的题目,更多时候,是让你用快速幂去加速一个递推式。
碰到这种题目,第一反应是根据我们对于高中数学中数列的理解,转化递推式。但是,在转换后,我们往往会得出一个含有无理数的式子,这种式子就不好用快速幂处理了
我们不妨以用时间复杂度算法求出斐波那契数列第
项这一问题为例
斐波那契递推式是
不难发现 该式子等价于
我们不妨将其转换成一个的矩阵乘上一个
的矩阵的矩阵乘法问题
给出代码
#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;
}