藍橋杯習題集_ 演算法訓練 矩陣乘法
阿新 • • 發佈:2019-01-01
原文:
演算法訓練 矩陣乘法 時間限制:1.0s 記憶體限制:512.0MB問題描述 輸入兩個矩陣,分別是m*s,s*n大小。輸出兩個矩陣相乘的結果。輸入格式 第一行,空格隔開的三個正整數m,s,n(均不超過200)。接下來m行,每行s個空格隔開的整數,表示矩陣A(i,j)。
接下來s行,每行n個空格隔開的整數,表示矩陣B(i,j)。輸出格式 m行,每行n個空格隔開的整數,輸出相乘後的矩陣C(i,j)的值。樣例輸入2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1樣例輸出-3 2
-8 2
提示
矩陣C應該是m行n列,其中C(i,j)等於矩陣A第i行行向量與矩陣B第j列列向量的內積。
例如樣例中C(1,1)=(1,0,-1)*(0,1,3) = 1 * 0 +0*1+(-1)*3=-3
AC程式碼:
#include<iostream> using namespace std; const int maxn=200; int a[maxn][maxn]; int b[maxn][maxn]; int c[maxn][maxn]; int main() { int i,j,k,m,s,n; cin>>m>>s>>n; for(i=0;i<m;i++) for(j=0;j<s;j++) cin>>a[i][j]; for(i=0;i<s;i++) for(j=0;j<n;j++) cin>>b[i][j]; for(i=0;i<m;i++) { for(j=0;j<n;j++) { c[i][j]=0; for(k=0;k<s;k++) { c[i][j]+=a[i][k]*b[k][j]; } } } for(i=0;i<m;i++) { for(j=0;j<n;j++) { cout<<c[i][j]<<" "; } cout<<endl; } return 0; }