矩阵快速幂与快速幂有异曲同工之处:
快速幂:
int ksm(int n,int k)//n的k次方
{
int ans=1;
while(k)
{
if(k&1)
{
ans*=n;
}
k/=2;
n*=n;
}
return ans;
}
矩阵乘法法则:(百度的,懒得自己写了)
现在,先打出乘法的模板。(三乘三模板)一般以结构体存储
node jzc(node a,node b)
{
node c;
for(int i=1; i<=3; i++)
{
for(int j=1; j<=3; j++)
{
c.m[i][j]=0;
for(int r=1; r<=3; r++)
{
c.m[i][j]+=a.m[r][j]*b.m[i][r];
}
}
}
return c;
}
现在出一个例子 1 2 3 的k次幂;
4 5 6
7 8 9
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
struct node
{
int m[10][10];
};
node jzc(node a,node b)
{
node c;
for(int i=1; i<=3; i++)
{
for(int j=1; j<=3; j++)
{
c.m[i][j]=0;
for(int r=1; r<=3; r++)
{
c.m[i][j]+=a.m[r][j]*b.m[i][r];
}
}
}
return c;
}
node jzm(node a,node b,int k)
{
while(k)
{
if(k&1)
{
b=jzc(a,b);
}
k/=2;
a=jzc(a,a);
}
return b;
}
int main()
{
int k;
scanf("%d",&k);
node a,b,c;
memset(a.m,0,sizeof(a.m));
memset(b.m,0,sizeof(b.m));
for(int i=1;i<=3;i++)
{
b.m[i][i]=1;
}
a.m[1][1]=1,a.m[1][2]=2,a.m[1][3]=3;
a.m[2][1]=4,a.m[2][2]=5,a.m[2][3]=6;
a.m[3][1]=7,a.m[3][2]=8,a.m[3][3]=9;
c=jzm(a,b,k);
for(int i=1;i<=3;i++)
{
for(int j=1;j<=3;j++)
{
printf("%d ",c.m[i][j]);
}
puts("");
}
}
这里的node b 相当于快速幂中的1;
可以发现和快速幂相同,只是需要手打乘法,再调用矩阵乘法。