藍橋杯-算法訓練-矩陣乘法-java
問題描述
輸入兩個矩陣，分別是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
思考
在第一個矩陣的列數和第二個矩陣的行數相等時才能實現矩陣的相乘，這里我定義二維數組來表示矩陣，通過for循環來進行矩陣的相乘。
代碼
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(); } }}
