您的位置:首页 > 其它

L1-048 矩阵A乘以矩阵B

2018-03-28 16:40 211 查看
给定两个矩阵A和B,要求你计算它们的乘积矩阵AB。需要注意的是,只有规模匹配的矩阵才可以相乘。即若A有Ra行、Ca列,B有Rb行、Cb列,则只有Ca与Rb相等时,两个矩阵才能相乘。输入格式:输入先后给出两个矩阵A和B。对于每个矩阵,首先在一行中给出其行数R和列数C,随后R行,每行给出C个整数,以1个空格分隔,且行首尾没有多余的空格。输入保证两个矩阵的R和C都是正数,并且所有整数的绝对值不超过100。输出格式:若输入的两个矩阵的规模是匹配的,则按照输入的格式输出乘积矩阵AB,否则输出“Error: Ca != Rb”,其中Ca是A的列数,Rb是B的行数。
输入样例1:
2 3
1 2 3
4 5 6
3 4
7 8 9 0
-1 -2 -3 -4
5 6 7 8
输出样例1:
2 4
20 22 24 16
53 58 63 28
输入样例2:
3 2
38 26
43 -5
0 17
3 2
-11 57
99 68
81 72
输出样例2:
Error: 2 != 3

注意:题目要求行末不能有空格,所以在最后输出的时候要注意输出格式
         三重循环运行超时
import java.util.Scanner;

public class Main {
public static void main(String[] args){
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int m = scanner.nextInt();
int array1[][] = new int
[m];
for(int i = 0; i < n; i++){
for(int j = 0; j < m; j++){
array1[i][j] = scanner.nextInt();
}
}
int p = scanner.nextInt();
int q = scanner.nextInt();
int array2[][] = new int[p][q];
int array[][] = new int
[q];
for(int i = 0; i < p; i++){
for(int j = 0; j < q; j++){
array2[i][j] = scanner.nextInt();
}
}
if(m != p){
System.out.println("Error: " + m + " != " + p);
}else{
for(int i = 0; i < n; i++){
for(int j = 0; j < q; j++){
for(int k = 0; k < m; k++){
array[i][j] += array1[i][k] * array2[k][j];
}
}
}
System.out.println(n + " " + q);
for(int i = 0; i < n; i++){
for(int j = 0; j < q; j++){
if(j != q - 1){
System.out.print(array[i][j] + " ");
}else{
System.out.println(array[i][j]);
}
}
}
}
scanner.close();
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: