藍橋杯 算法訓練 矩陣乘法 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();
	}	
}