問題 F: 矩陣乘法

題目描述

請你實現一個程式，用於求兩個矩陣的乘積。

輸入

包括一系列的測試用例，每個測試用例的第一行包含三個整數a，b，c，其中a是第一個矩陣的行數，b是第一個矩陣的列數並且是第二個矩陣的行數，c是第二個矩陣的列數，接下來是a行，每行包含b個整數，每個整數用空格隔開，用於表示第一個矩陣；然後是b行，每行包含c個整數，每個整數用空格隔開，用於表示第二個矩陣。

輸出

對應每個測試用例，輸出a行，每行包括c個整數，每個整數用空格隔開，用於表示兩個矩陣乘積後的a行c列矩陣。

樣例輸入

2 3 4

1 2 3

4 5 6

1 2 3 4

5 6 7 8

9 10 11 12

樣例輸出

38 44 50 56

83 98 113 128

import java.util.Scanner;
public class  F{
	public static void main(String[] args) {
		 Scanner in = new Scanner(System.in);
		 String first[];
	     int a,b,c;
	     while(in.hasNext())           //沒說按0結束都要這樣判斷
	     {    
	    	 first = (in.nextLine()).split(" ");
             a = Integer.parseInt(first[0]);
             b = Integer.parseInt(first[1]);
             c = Integer.parseInt(first[2]);
             //System.out.println(a+" "+b+" "+c);
             String[][] arr1 = new String[a][b];
             String[][] arr2 = new String[b][c];
             for(int i=0;i<a;i++)   //輸入第一個矩陣a行b列
             {
            	 arr1[i]=(in.nextLine()).split(" ");
             }
             for(int j=0;j<b;j++)   //輸入第二個矩陣b行c列
             {
            	 arr2[j]=(in.nextLine()).split(" ");
             }
             int[][] arr3 = multiplyMatrix(arr1,arr2);
             printMatrix(arr3); 
	     }
	}
	private static int[][] multiplyMatrix(String[][] arr1, String[][] arr2) 
	{
		if(arr1[0].length != arr2.length){
			return null;
		}
			int[][] c=new int[arr1.length][arr2[0].length];
		for(int i=0;i<arr1.length;i++) {            //拿出arr1的每一行
			for(int j=0;j<arr2[0].length;j++) {      //拿出arr2的每一列  
			  for(int k=0;k<arr1[0].length;k++) { //k表示arr1一行有多少個數
			             c[i][j] += Integer.parseInt(arr1[i][k]) * Integer.parseInt(arr2[k][j]);             //一行乘以一列的和就是矩陣的一個值
			         }
			     }
			 }
			return c;
	}
	
	private static void printMatrix(int[][] arr3) {
			if(arr3!=null){
			for(int i=0;i<arr3.length;i++){             //行號
				for(int j=0;j<arr3[0].length;j++){      //列號
					if(j!=arr3[0].length-1)
						System.out.printf("%d ",arr3[i][j]);
					else
						System.out.printf("%d",arr3[i][j]);
				}
				System.out.println();
			}
		}
	}
}