蓝桥杯 算法训练 矩阵乘法 java

试题 算法训练 矩阵乘法

资源限制
时间限制：1.0s 内存限制：512.0MB
问题描述
　　输入两个矩阵，分别是ms，sn大小。输出两个矩阵相乘的结果。
输入格式
　　第一行，空格隔开的三个正整数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 +01+(-1)*3=-3

//解题思路：了解矩阵的相乘公式，并提出题中“两个矩阵，分别是m*s，s*n大小”
package Main;
import java.util.Scanner;
public class Main {
	public static void main(String[] args) {   
      Scanner sc=new Scanner(System.in);  
       int m=sc.nextInt();
       int s=sc.nextInt();
       int n=sc.nextInt();
       //ar数组为接收m*s数列，br接收s*n数列，ba接收最后的输出数据
       int[][] ar=new int[m][s];
       int[][] br=new int[s][n];
       int[][] ab=new int[m][n];
       for (int i = 0; i < m; i++) {
		for (int j = 0; j < s; j++) {
			ar[i][j]=sc.nextInt();
			
		}
		
	}
       for (int i = 0; i <s; i++) {
   		for (int j = 0; j < n; j++) {
   			br[i][j]=sc.nextInt();
   			
   		}
   		
   	}
   	//进行乘法的计算
       for (int i = 0; i < m; i++) {
   		for (int j = 0; j < n; j++) {
   			int sum=0;
   			for (int j2 = 0; j2 < s; j2++) {
				sum=ar[i][j2]*br[j2][j]+sum;				
			}
   			ab[i][j]=sum;
   			
   		}
   		
   	}
       for (int i = 0; i < m; i++) {		
		for (int j = 0; j < n; j++) {
			System.out.print(ab[i][j]+" ");
			
		}
		System.out.println();
	}
    sc.close();
	}	
}