洛谷 P3390 矩阵快速幂模板

题目背景
矩阵快速幂
题目描述
给定n*n的矩阵A,求A^k
输入输出格式
输入格式:
第一行,n,k
第2至n+1行,每行n个数,第i+1行第j个数表示矩阵第i行第j列的元素
输出格式:
输出A^k
共n行,每行n个数,第i行第j个数表示矩阵第i行第j列的元素,每个元素模10^9+7
输入输出样例
输入样例#1:
2 1
1 1
1 1
输出样例#1:
1 1
1 1
说明

n<=100, k<=10^12, |矩阵元素|<=1000 算法:矩阵快速幂

吐槽:wa了两次太丢人了,第一次发现没有取模,第二次发现没有开long long,简直了,QAQ。

#include 
#include 
#include 
#include 
#define mod 1000000007
using namespace std;
int n;
long long m;
struct node
{
	int cal,line;
	long long a[100][100];
};
node pow(node x,node y)
{
	node re;
	memset(re.a,0,sizeof(re.a));
	re.cal=x.cal;re.line=y.line;
	for(int i=1;i<=x.cal;i++)
	    for(int k=1;k<=y.line;k++)
	        for(int j=1;j<=x.line;j++)
	          re.a[i][k]=(re.a[i][k]+(x.a[i][j]*y.a[j][k])%mod)%mod;
	return re;
}
int main()
{	

        node tmp,ans;
	scanf("%d%lld",&n,&m);
	for(int i=1;i<=n;i++)
	   for(int j=1;j<=n;j++)
	       scanf("%lld",&tmp.a[i][j]);
	for(int i=1;i<=n;i++)
	    ans.a[i][i]=1;
	ans.cal=n;ans.line=n;tmp.cal=n;tmp.line=n;
	while(m)
	{
		if(m&1) ans=pow(ans,tmp);
		tmp=pow(tmp,tmp);
		m>>=1;
	}
	for(int i=1;i<=ans.cal;i++)
	{
		for(int j=1;j


你可能感兴趣的:(数论——快速幂)