矩阵快速幂（重载运算符，矩阵乘法）

上一篇写了快速幂[模板] ，这篇是有关拓展，矩阵快速幂=矩阵乘法+快速幂。下面是洛谷里的一道模板题，写之前要了解矩阵乘法。关于重载运算符，这里用不用都行。

【模板】矩阵快速幂

题目描述

给定 $n\times n$ 的矩阵 $A$，求 $A^k$。

输入格式

第一行两个整数 $n,k$。
接下来 $n$ 行，每行 $n$ 个整数，第 $i$ 行的第 $j$ 的数表示 $A_{i,j}$。

输出格式

输出 $A^k$

共 $n$ 行，每行 $n$ 个数，第 $i$ 行第 $j$ 个数表示 $(A^k{)}_{i,j}$，每个元素对 $10^9+7$ 取模。

【数据范围】

对于 $100\%$ 的数据，$1\le n\le 100$，$0\le k\le 10^{12}$，$|A_{i,j}|\le 1000$。

代码

#include<iostream>
#include<cstring>
#define ll long long 
const ll N=1e9+7;
ll n;
using namespace std;
typedef struct {
	ll a[105][105];
	void juzhen(){
			memset(a,0,sizeof a);//结构体赋值
	}
       void unt()//创建单位矩阵
	{
		for(ll i=1;i<=n;i++)
			a[i][i]=1;
	}
}juzhen;
juzhen operator*(const juzhen &d,const juzhen &b)//重载运算符，进行矩阵乘法
{
	juzhen c;
	memset(c.a,0,sizeof c.a);//很重要，明明结构体里已经赋为零了，但这里不加就错
for(ll k=1;k<=n;++k)
		for(ll i=1;i<=n;++i)
			for(ll j=1;j<=n;++j)
				c.a[i][j]=(c.a[i][j]+d.a[i][k]*b.a[k][j]%N)%N;
 
	return c;
}
signed main()
{
	ll k,t;
	cin>>n>>k;
	juzhen a;
	for(ll i=1;i<=n;i++)
	for(ll j=1;j<=n;j++)
	{
		cin>>t;
		a.a[i][j]=t;
	}
	juzhen ans;ans.unt();//建立一个单位矩阵，与任何矩阵相乘，等于矩阵本身，相当于一次幂
	while(k)
	{ 
		if(k&1) ans=ans*a;//与快速幂思想一样，为奇数时结果上*a;但*不能缩写为*=
		a=a*a;k>>=1;
	}
	for(ll i=1;i<=n;putchar('\n'),i++)
	{
		for(ll j=1;j<=n;j++)
			cout<<ans.a[i][j]<<" ";
	}
		return 0;
}

这道题真是废了老大劲，果然不能在没状态时敲，漏洞百出，花了好长时间补bug。我是看了洛谷上的题解，边理解边照猫画虎，也可以去那里看看