首页 > 代码库 > 矩阵乘法 洛谷 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【模板】矩阵快速幂
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。