【总结】矩阵快速幂

在学习矩阵快速幂之前,首先我们需要分别了解快速幂和矩阵乘法

快速幂

快速幂要求解的是这样一类问题:

给你A,B,C,求A的B次方模C的余数

A,C<=109,B<=1018

如果我们线性去求,时间复杂度是O(n)的,但题目中给出的B是很大的数,这样显然会超时,我们可以用快速幂来加速这个过程。

我们可以想像一下小学的时候我们如何计算2^16

216=48=164=2562=65536

那如何计算2^18呢?

218=49=448=4*164=4256^2=4*65536=262144

快速幂同理也是如此

我们可以按照上面做法,利用分治的思想求去解

这样原本O(n)的时间复杂度便降到了O(log n )

long long ans=1,base=a;
while(n>0){
		if(n&1){
			ans*=base;
		}
		base*=base;
		n=n/2;
	}

矩阵乘法

矩阵乘法可以先稍作了解,知道矩阵相乘的运算法则

\(C[i][j]= A[i][k]B[k][j]\)

矩阵快速幂

矩阵快速幂的原理同快速幂一样,只是转换为了矩阵之间的乘法操作

所以单纯的重载一下运算符,将普通的乘法转换为矩阵乘法就好了。

嗯,看一下代码就应该很好理解了x

#include<iostream>
#include<cstdio>
#include<cctype>
#define ll long long
#define Mod 1000000007
using namespace std;
ll read(){
	ll a=0;int f=0;char p=getchar();
	while(!isdigit(p)){f|=p=='-';p=getchar();}
	while(isdigit(p)){a=(a<<3)+(a<<1)+(p^48);p=getchar();}
	return f?-a:a;
}
ll n,k;
struct mat{
	ll m[101][101];
}a,b,c,e;
mat mul(mat x,mat y){
	mat k;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			k.m[i][j]=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			for(int q=1;q<=n;q++)
				k.m[i][j]=k.m[i][j]%Mod+x.m[i][q]%Mod*y.m[q][j]%Mod;
	return k;
}
mat pow(mat x,ll y){
	mat ans=b;
	while(y){
		if(y&1)ans=mul(ans,x);
		x=mul(x,x);
		y>>=1;
	}
	return ans;
}
int main(){
	n=read();k=read();
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			a.m[i][j]=read();
	for(int i=1;i<=n;i++)b.m[i][i]=1;
	mat ans=pow(a,k);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++)
			cout<<ans.m[i][j]%Mod<<" "; 
		cout<<endl;
	}
}
posted @ 2020-01-17 21:39  蕙心心w  阅读(224)  评论(0编辑  收藏  举报