矩阵快速幂(模板)

矩阵快速幂模板
传送门

#include <bits/stdc++.h>
using namespace std;
long long n;
struct mat
{
	long long g[110][110];
	
	void clear()
	{
		memset (g, 0, sizeof (g));//清0
		return ;
	}
	
	mat operator * (const mat& b)
	{
		mat  tmp;//tmp要作为返回值
		tmp.clear();//因为要累加,所以要事先清0
		for (long long i = 0; i < n; i ++)//基本枚举,下同
			for (long long j = 0; j < n; j ++)
				for (long long k = 0; k < n; k ++)
					tmp.g[i][j] += (g[i][k]*b.g[k][j])%1000000007, tmp.g[i][j] %= 1000000007;
		return tmp;
	}	
	void print()
	{
		for (long long i = 0; i < n; i ++, printf ("\n"))
			for (long long j = 0; j < n; j ++)
				printf ("%lld ", g[i][j]);
		return ;//个人喜欢把输入输出封装一下,下同
	}
	
	void scan()
	{
		for (long long i = 0; i < n; i ++)
			for (long long j = 0; j < n; j ++)
				scanf ("%lld", &g[i][j]);
		return ;
	}
};

mat a;

mat mypow(long long k)
{
	mat res;
    res.clear();
	for (long long i = 0; i < n; i ++)
		res.g[i][i] = 1;
	while (k)
	{
		if (k%2)
			res =res*a;
		k >>=1;
		a =a*a;
	}
	return res;
}
long long k;
int main()
{
    cin>>n>>k;
    a.scan();
    mat res=mypow(k);
    res.print();
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值