Java教程

矩阵快速幂

本文主要是介绍矩阵快速幂,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

在这里插入图片描述

  1. 初始化单元矩阵
  2. 输入矩阵数值
  3. 类似于快速幂求解矩阵乘法
  4. 重载运算法*,用来表示矩阵乘法
#include<bits/stdc++.h>
using namespace std;
const int maxn=110;
const int mod=1e9+7;
struct mat{
	int m[maxn][maxn];
}unit;
int n;
void init_unit()
{
	for(int i=0;i<maxn;i++)
		unit.m[i][i]=1;
	return ;
}
mat operator *(mat a,mat b)
{
	mat ret;
	long long x;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			x=0;
			for(int k=0;k<n;k++)
				x=(x%mod+((long long)a.m[i][k]*b.m[k][j])%mod)%mod;
			ret.m[i][j]=x%mod;
		}
	}
	return ret;
}
mat pow_mat(mat a,long long y)
{
	mat res=unit;
	while(y)
	{
		if(y&1) res=res*a;
		y>>=1;
		a=a*a;
	}
	return res;
}
int main()
{
	long long x;
	init_unit();
	while(cin>>n>>x)
	{
		mat a;
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				cin>>a.m[i][j];
		a=pow_mat(a,x);
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				if(j+1==n) cout<<a.m[i][j]<<endl;
				else cout<<a.m[i][j]<<" ";
	}
	return 0;
}
这篇关于矩阵快速幂的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!