1. 程式人生 > >藍橋杯習題集_ 演算法訓練 矩陣乘法

藍橋杯習題集_ 演算法訓練 矩陣乘法

原文:

演算法訓練 矩陣乘法  時間限制: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;
}