首页 > 代码库 > 矩阵乘法 洛谷 P3390【模板】矩阵快速幂

矩阵乘法 洛谷 P3390【模板】矩阵快速幂

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 算法:矩阵快速幂

 

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 const long long mod=1000000007;
 7 int n;
 8 long long k;
 9 struct matrix{
10     long long a[110][110];
11     matrix operator*(const matrix &b)const{
12         matrix ret;
13         for(int i=1;i<=n;i++)
14             for(int j=1;j<=n;j++){
15                 ret.a[i][j]=0;
16                 for(int k=1;k<=n;k++) ret.a[i][j]=(ret.a[i][j]+a[i][k]*b.a[k][j])%mod;
17             }
18         return ret;
19     }
20 }a;
21 matrix matrix_pow(matrix a,long long b){
22     matrix ret=a,tmp=a;
23     b--;
24     for(;b;b>>=1,tmp=tmp*tmp)//!!!
25         if(b&1) ret=ret*tmp;
26     return ret;
27 }
28 int main(){
29     scanf("%d%lld",&n,&k);
30     for(int i=1;i<=n;i++)
31         for(int j=1;j<=n;j++) scanf("%d",&a.a[i][j]);
32     a=matrix_pow(a,k);
33     for(int i=1;i<=n;i++){
34         for(int j=1;j<n;j++) printf("%d ",a.a[i][j]);
35         printf("%d\n",a.a[i][n]);
36     }
37     return 0;
38 }

 

矩阵乘法 洛谷 P3390【模板】矩阵快速幂