基准时间限制:3 秒 空间限制:131072 KB 分值: 40
难度:4级算法题
给出一个N * N的矩阵,其中的元素均为正整数。求这个矩阵的M次方。由于M次方的计算结果太大,只需要输出每个元素Mod (10^9 + 7)的结果。
Input
第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)
Output
共N行,每行N个数,对应M次方Mod (10^9 + 7)的结果。
Input示例
2 3 1 1 1 1
Output示例
4 4 4 4
思路:
1、矩阵快速幂的裸题啊。
然而后台数据中包含0的情况很少啊...黑科技优化度为0啊.....23333333
不过还是介绍一下,矩阵乘法优化的黑科技:http://blog.csdn.net/mengxiang000000/article/details/51932183
Ac代码(跑的好慢啊):
#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
#define ll __int64
ll mod=1000000007;
ll n,m;
typedef struct Matrix
{
ll mat[101][101];
}matrix;
matrix A,B;
Matrix matrix_mul(matrix a,matrix b)
{
matrix c;
memset(c.mat,0,sizeof(c.mat));
int i,j,k;
for(int i=0;i<n;i++)
{
for(int k=0;k<n;k++)
{
if(a.mat[i][k]==0)continue;
for(int j=0;j<n;j++)
{
c.mat[i][j]=c.mat[i][j]+(a.mat[i][k]*b.mat[k][j])%mod;
c.mat[i][j]%=mod;
}
}
}
return c;
}
Matrix matrix_quick_power(matrix a,ll k)
{
matrix b;
memset(b.mat,0,sizeof(b.mat));
for(int i=0;i<n;i++)
b.mat[i][i]=1;//单位矩阵b
while(k)
{
if(k%2==1)
{
b=matrix_mul(a,b);
k-=1;
}
else
{
a=matrix_mul(a,a);
k/=2;
}
}
return b;
}
int main()
{
while(cin>>n>>m)
{
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%I64d",&A.mat[i][j]);
}
}
B=matrix_quick_power(A,m);
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
printf("%I64d ",B.mat[i][j]%mod);
}
printf("\n");
}
}
}