1113 矩阵快速幂
基准时间限制: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
/*
代码是看的大佬的博客写出来的,就先当模板来理解吧
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
using namespace std;
typedef long long ll;
ll mod=1e9+7;// 相当于宏定义 mod
int n;
typedef struct node//定义结构体,为接下来的代码设置矩阵方便许多
{
ll a[101][101];
}matrix;
matrix mat_mul(matrix p,matrix q)//返回的是结构体类型的矩阵,所以定义成结构体类型的被调函数
{
matrix res;//结果矩阵
int i,j,k;
ll x;
memset(res.a,0,sizeof(res.a));
for(i=0;i<=n-1;i++){ //矩阵乘法
for(j=0;j<=n-1;j++){
x=0;
for(k=0;k<=n-1;k++){
x+=(p.a[i][k]*q.a[k][j]);
x%=mod; //先求完 和 之后在 mod
}
res.a[i][j]=x%mod; 防止数据超出范围,所以在 mod 一次
}
}
return res; //返回矩阵
}
matrix quick_pow(matrix mat,ll m)//矩阵快速幂的模板
{
matrix unit;
int i;
memset(unit.a,0,sizeof(unit.a));
for(i=0;i<=n-1;i++)
unit.a[i][i]=1;//设置单位矩阵,主对角线都是 1,其余是 0
while(m){
if(m&1)
unit=mat_mul(unit,mat);//相当于 ans=mat_mul(E,mat);
mat=mat_mul(mat,mat); // mat=mat_mul(mat,mat); 顺序不能变
m>>=1;
}
return unit;//返回结果
}
int main()
{
int i,j;
ll m;
scanf("%d %lld",&n,&m);
matrix s,ans;
for(i=0;i<=n-1;i++)
for(j=0;j<=n-1;j++)
scanf("%lld",&s.a[i][j]);
ans=quick_pow(s,m);
for(i=0;i<=n-1;i++){
for(j=0;j<=n-1;j++){
if(j<n-1)
printf("%lld ",ans.a[i][j]);
else
printf("%lld\n",ans.a[i][j]);
}
}
}