蓝桥杯寒假任务之矩阵乘法

有什么比一次AC更快乐的事呢？作为今年寒假刚刚学习编程的大菜鸟，我的天，我竟然一次AC了。不过这个代码花的时间很多，刚好是限制时间1秒。需要优化。不过代码长度比其他的代码短一点。

问题描述

　　输入两个矩阵，分别是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

import java.util.Scanner;
public class Main {


public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner input = new Scanner(System.in);
int m = input.nextInt();
int n = input.nextInt();
int k = input.nextInt();
int a[][] = new int[m][n];
int b[][] = new int[n][k];
int c[][] = new int[m][k];

for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
a[i][j] = input.nextInt();
}
}

for(int j=0;j<n;j++){
for(int r=0;r<k;r++){
b[j][r] = input.nextInt();
}
}

for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
for(int r=0;r<k;r++){
c[i][r]+=a[i][j]*b[j][r];
}
}
}

for(int i=0;i<m;i++){
for(int r=0;r<k;r++){
System.out.print(c[i][r]+" ");
}
System.out.println();
}
}
 }

写完看了别人的代码，发现这个代码，用时比较短，可能是因为少了一个for循环吧。仔细看了看，厉害了我的哥！！

Noedge的import java.util.Scanner;public class Main { public static void main(String[] args) { Scanner scanner = new Scanner(System.in); int m = 0 , s = 0 , n = 0 ; if(scanner.hasNext()){ //输入m,s,n m = scanner.nextInt() ; s = scanner.nextInt() ; n = scanner.nextInt() ; } //定义两个矩阵,行列数分别为m,s与s,n int arr1[][] = new int[m][s] ; int arr2[][] = new int[s][n] ; for(int i = 0 ; i < m ; i ++){ //输入第一个矩阵 for(int j = 0 ; j < s ; j++){ if(scanner.hasNext()){ arr1[i][j] = scanner.nextInt() ; } } } for(int i = 0 ; i < s ; i++){ //输入第二个矩阵 for(int j = 0 ; j < n ; j++){ if(scanner.hasNext()){ arr2[i][j] = scanner.nextInt() ; } } } int arr[][] = new int[m][n] ; //定义一个相乘后的矩阵，矩阵的行数与列数为m，n for(int i = 0 ; i < m ; i++){ //进行矩阵相乘 for(int j = 0 ; j < n ; j++){ for(int x = 0 ; x < s ; x ++){ //arr[i][j] = arr1[i][1] * arr2[1][j] + ... + arr1[i][s-1] * arr2[s-1][j] arr[i][j] += arr1[i][x] * arr2[x][j] ; } System.out.print(arr[i][j] + " "); } System.out.println(); } }}