A*B Problem II
时间限制:
1000 ms | 内存限制:
65535 KB
难度:
1
-
描述
-
ACM的C++同学有好多作业要做,最头痛莫过于线性代数了,因为每次做到矩阵相乘的时候,大量的乘法都会把他搞乱,所以他想请你写个程序帮他检验一下计算结果是否正确。
-
输入
-
有多组测试数据,每行给出一组m,n,k(0<m,n,k<=50)。m,n,k表示两个矩阵的大小,其中:
矩阵A:m行n列。
矩阵B:n行k列。
接下来给出m*n个数表示矩阵A和n*k个数表示矩阵B,对于每个数s,0<=s<=1000。
当m,n,k同时为0时结束。
输出
- 计算两个矩阵的乘积并输出。 样例输入
-
2 1 3 1 2 1 2 3 2 2 3 1 2 3 4 1 0 1 0 1 0 0 0 0
样例输出
-
1 2 3 2 4 6 1 2 1 3 4 3
-
-
有多组测试数据,每行给出一组m,n,k(0<m,n,k<=50)。m,n,k表示两个矩阵的大小,其中:
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while (scanner.hasNext()) {
int row = scanner.nextInt();
int mid = scanner.nextInt();
int col = scanner.nextInt();
int matrix1[][] = new int[row][mid];
int matrix2[][] = new int[mid][col];
long result[][] = new long[row][col];
for (int i = 0; i < row; i++) {
for (int j = 0; j < mid; j++) {
matrix1[i][j] = scanner.nextInt();
}
}
for (int i = 0; i < mid; i++) {
for (int j = 0; j < col; j++) {
matrix2[i][j] = scanner.nextInt();
}
}
for (int i = 0; i < row; i++) {
for (int j = 0; j < mid; j++) {
for (int k = 0; k < col; k++) {
result[i][k] += matrix1[i][j] * matrix2[j][k];
}
}
}
for (int i = 0; i < row; i++) {
for (int j = 0; j < col; j++) {
System.out.print(result[i][j] + " ");
}
System.out.println();
}
}
}
}