- 首先给出矩阵乘法的一般写法(矩阵快速幂只用到方阵,所以这里只给出了方阵的语法,一般写法只需要改以下循环的条件就行了)
const int N=100;
int c[N][N];
void multi(int a[][N],int b[][N],int n)//n是矩阵大小,n<N
{
memset(c,0,sizeof c);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
for(int k=1;k<=n;k++)
c[i][j]+=a[i][k]*b[k][j];
}
- 矩阵快速幂算法(将快速幂中的乘法改成矩阵乘法即可)
const int N=10;
int tmp[N][N];
void multi(int a[][N],int b[][N],int n)
{
memset(tmp,0,sizeof tmp);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
for(int k=0;k<n;k++)
tmp[i][j]+=a[i][k]*b[k][j];
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
a[i][j]=tmp[i][j];
}
int res[N][N];
void Pow(int a[][N],int n)
{
memset(res,0,sizeof res);//n是幂,N是矩阵大小
for(int i=0;i<N;i++) res[i][i]=1;
while(n)
{
if(n&1)
multi(res,a,N);//res=res*a;复制直接在multi里面实现了;
multi(a,a,N);//a=a*a
n>>=1;
}
}
快速幂算法的关键在下面这段代码
while(n)
{
if(n&1)
multi(res,a,N);//res=res*a;复制直接在multi里面实现了;
multi(a,a,N);//a=a*a
n>>=1;
}
- 题目实战
(1)斐波那契数列(高阶)
原理如下
有了这个递推公式,计算时的重点就变成了如何更快的计算n阶矩阵{(1,1)(1,0)}的值。具体代码只用在矩阵快速幂的基础上稍加改动即可。 - 关于这一算法的更具体的解析和其他实例,欢迎大家补充,我也会在之后不断完善。