51nod 1113 矩阵快速幂 （简单模板）

题目链接：

http://www.51nod.com/Challenge/Problem.html#!#problemId=1113

题目描述：

1113 矩阵快速幂

给出一个N * N的矩阵，其中的元素均为正整数。求这个矩阵的M次方。由于M次方的

计算结果太大，只需要输出每个元素Mod (10^9 + 7）的结果。

输入

第1行：2个数N和M，中间用空格分隔。N为矩阵的大小，M为M次方。(2 <= N <= 100, 1 <= M <= 10^9)
第2 - N + 1行：每行N个数，对应N * N矩阵中的1行。(0 <= N[i] <= 10^9)

输出

共N行，每行N个数，对应M次方Mod (10^9 + 7)的结果。

输入样例

2 3
1 1
1 1

输出样例

4 4
4 4

思路：

       矩阵快速幂的简单模板。

代码实现：

#include<bits/stdc++.h>
#define LL long long
using namespace std;
const int N=2e5+100;
const int mod=1e9+7;
int a[101][101],f[101][101];
int tmp[101][101];
void  mul(int aa[101][101],int bb[101][101],int n)
{
	memset(tmp,0,sizeof(tmp));
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			for(int k=1;k<=n;k++)
				tmp[i][j]=(tmp[i][j]+(LL)aa[i][k]*bb[k][j])%mod;
	memcpy(aa,tmp,sizeof(tmp));	
}
int main()
{
	int n,m;
	while(cin>>n>>m)
	{
		memset(f,0,sizeof(f));
		for(int i=1;i<=n;++i)
			for(int j=1;j<=n;++j)cin>>a[i][j];
		for(int i=1;i<=n;++i)f[i][i]=1;
		while(m)
		{
			if(m&1)mul(f,a,n);
			mul(a,a,n);
			m>>=1;
		}
		int count=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
				cout<<f[i][j]<<" ";
			cout<<endl;
		}
		
	}
	return 0;
}

The end；