洛谷p3390 矩阵快速幂模板

题目链接
存一下矩阵快速幂的板子。= =
AC代码:

#include
using namespace std;
typedef long long ll;
const int maxn=105; 
const int mo=1e9+7;
ll n,m;
struct ahaha
{
	ll a[maxn][maxn];     //一定要用long long存矩阵,否则在过程中会爆掉
	ahaha()
	{
		memset(a,0,sizeof a);
	}
	inline void build()
	{     //建造单位矩阵
		for(int i=1;i<=n;++i)a[i][i]=1;
	}
}a;
ahaha operator *(const ahaha &x,const ahaha &y)
{     //重载运算符
	ahaha z;
	for(int k=1;k<=n;++k)
		for(int i=1;i<=n;++i)
			for(int j=1;j<=n;++j)
				z.a[i][j]=(z.a[i][j]+x.a[i][k]*y.a[k][j]%mo)%mo;
	return z;
}
ll k;
ahaha qk(ahaha a,ll b)
{
	ahaha ans;
	ans.build();
	while(b)
	{
		if(b&1) ans=ans*a;
		a=a*a;
		b=b>>1;
	}
	return ans;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			scanf("%lld",&a.a[i][j]);
		}
	}
	ahaha fin=qk(a,m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(j!=n) printf("%lld ",fin.a[i][j]);
			else printf("%lld\n",fin.a[i][j]);
		}
	}
	return 0;
}

你可能感兴趣的:(刷题记录,板子梳理)