矩阵乘法 Java实现

1. 矩阵乘法

计算两个矩阵的乘法。nm阶的矩阵A乘以mk阶的矩阵B得到的矩阵C 是n*k阶的，且C[i][j] = A[i][0]*B[0][j] + A[i][1]*B[1][j] + …… +A[i][m-1]*B[m-1][j](C[i][j]表示C矩阵中第i行第j列元素)。

输入

第一行为n, m, k，表示A矩阵是n行m列，B矩阵是m行k列，n, m, k均小于100
然后先后输入A和B两个矩阵，A矩阵n行m列，B矩阵m行k列，矩阵中每个元素的绝对值不会大于1000。

输出

输出矩阵C，一共n行，每行k个整数，整数之间以一个空格分开。

样例输入

3 2 3
1 1
1 1
1 1
1 1 1
1 1 1

样例输出

2 2 2
2 2 2
2 2 2

代码：

import java.util.Scanner;

/**
 * @author: gznc_pcc
 * @date:2019年4月24日21:38:50
 * @version :
 */
public class MatrixMultiplication {
	/**
	 * 矩阵乘法
	 * a点乘b，当矩阵a的列数x与矩阵b的行数y相等时可进行相乘
	 * a乘b得到的新矩阵c，c的行数y等于a的行数，c的列数x等于b的列数
	 **/
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		int n,m,k;
		int [][]a = new int[102][102];
		int [][]b = new int[102][102];
		int [][]c = new int[102][102];
		Scanner scanner = new Scanner(System.in); 
		n = scanner.nextInt(); 
		m = scanner.nextInt();
		k = scanner.nextInt();
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=m;j++) {
				a[i][j] = scanner.nextInt();
			}
		}
		for(int i=1;i<=m;i++) {
			for(int j=1;j<=k;j++) {
				b[i][j] = scanner.nextInt();
			}
		}
		c = matrix(a, b);
		for(int i=1;i<=n;i++){
			for(int j=1;j<=k;j++) {
				System.out.print(c[i][j]+" ");
			}
			System.out.println();
		}	
	}
	public static int[][] matrix(int a[][], int b[][]) {
        if (a[0].length != b.length)
            return null;
        int y = a.length;
        int x = b[0].length;
        int c[][] = new int[y][x];
        for (int i = 0; i < y; i++)
            for (int j = 0; j < x; j++)
                for (int k = 0; k < b.length; k++)
                    c[i][j] += a[i][k] * b[k][j];
        return c;
    }
}

结果截图：