矩阵的快速幂

矩阵的快速幂

前置知识:
1. 基本快速幂
2. c++运算符重载
3. 幺元基本概念(任何一个数乘以幺元等于本身)

#include"iostream"
using namespace std;
const int mod = 1e9 + 7;
struct mat{
	long long int a[110][110];
}unit;
int n;
mat operator * (mat v1, mat v2) //重载 乘法 运算符
{
	mat ret;
	long long int 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+(((v1.a[i][k]) %mod* (v2.a[k][j]%mod)) %mod);
			ret.a[i][j] = x % mod ;
		}
	return ret;  //记录结果
}
void init()
{
	for(int i = 0;i<n;i++)   //定义幺元
		unit.a[i][i] = 1;	//矩阵的幺元是单位矩阵
}
mat mat_pow(mat a,int n) //基本的快速幂模板
{
	mat res = unit;
	while(n)
	{
		if(n&1)
			res = res*a;
		a = a*a;
		n >>= 1;
	}
	return res;
}
int main()
{
	int m;
	cin>>n>>m;
	init();
	mat t1,t2;
	for(int i = 0;i<n;i++)
		for(int j = 0;j<n;j++)
			cin>>t1.a[i][j];

	mat tmp = mat_pow(t1,m);
	for(int i = 0;i<n;i++)
	{
		for(int j = 0;j<n;j++)
			cout<<tmp.a[i][j]<<" ";
		cout<<endl;
	}

	
	
}