C++矩阵乘法

题目描述

计算两个矩阵的乘法。n×m 阶的矩阵 A 乘以 m×k 阶的矩阵 B 得到的矩阵 C 是 n×k 阶的，且 C[i][j]=A[i][0]×B[0][j]+A[i][1]×B[1][j]+C[i][j]=A[i][0]×B[0][j]+A[i][1]×B[1][j]+ …… +A[i][m−1]×B[m−1][j](C[i][j]+A[i][m−1]×B[m−1][j](C[i][j] 表示 C 矩阵中第 i 行第 j 列元素）。

输入格式

第一行为 n,m,k表示 A 矩阵是 n 行 m 列，B 矩阵是 m 行 k 列，n,m,k均小于 100。

然后先后输入 A 和 B 两个矩阵，A 矩阵 n 行 m 列，B 矩阵 m 行 kk 列，矩阵中每个元素的绝对值不会大于 1000。

输出格式

输出矩阵 C，一共 n 行，每行 k 个整数，整数之间以一个空格分开。

输入输出样例

输入 #1

3 2 3
1 1
1 1
1 1
1 1 1
1 1 1

输出 #1

2 2 2
2 2 2
2 2 2

代码如下：

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n,m,k;
	int a[1001][1001],b[1001][1001],c[1001][1001];
	cin>>n>>m>>k;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			cin>>a[i][j];
		}
	}
	for(int i=0;i<m;i++){
		for(int j=0;j<k;j++){
			cin>>b[i][j];
		}
	}
	for(int ii=0;ii<n;ii++){
		for(int jj=0;jj<k;jj++){
			int s=0;
			for(int k=0;k<m;k++){
				s+=a[ii][k]*b[k][jj];
			}
			c[ii][jj]=s;
		}
	}
	for(int i=0;i<n;i++){
		for(int j=0;j<k;j++){
			cout<<c[i][j]<<" ";
		}
		cout<<endl;
	}
	return 0;
}