洛谷 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 <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#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<ans.line;j++)
	       printf("%lld ",ans.a[i][j]);
	    printf("%lld\n",ans.a[i][ans.line]);
	}
	return 0;
}