蓝桥杯 算法提高 矩阵相乘 Java

算法提高 矩阵相乘

这是一道基础题，通过自己写矩阵的i,j找出矩阵相乘下标规律，用循环直接完成即可。

问题描述
　　小明最近在为线性代数而头疼，线性代数确实很抽象（也很无聊），可惜他的老师正在讲这矩阵乘法这一段内容。
　　当然，小明上课打瞌睡也没问题，但线性代数的习题可是很可怕的。
　　小明希望你来帮他完成这个任务。

现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵，
　　要你求出他们相乘的积（当然也是矩阵）。
　　(输入数据保证aj=bi,不需要判断)
输入格式
　　输入文件共有ai+bi+2行，并且输入的所有数为整数（long long范围内）。
　　第1行：ai 和 aj
　　第2~ai+2行：矩阵a的所有元素
　　第ai+3行：bi 和 bj
　　第ai+3~ai+bi+3行：矩阵b的所有元素
输出格式
　　输出矩阵a和矩阵b的积（矩阵c)
　　（ai行bj列）
样例输入
2 2
12 23
45 56
2 2
78 89
45 56
样例输出
1971 2356
6030 7141

import java.util.*;
public class Main {
	static ArrayList<Integer> al = new ArrayList<Integer>();
	
    public static void main(String[] args) {
    	Scanner sn = new Scanner(System.in);
    	
    	int n1 = sn.nextInt();
    	int m1 = sn.nextInt();
    	long [][]matrix1 = new long[n1][m1];
    	for(int i = 0;i < n1;i++)
    		for(int j = 0;j < m1;j++)
    			matrix1[i][j] = sn.nextLong();
    	
    	int n2 = sn.nextInt();
    	int m2 = sn.nextInt();
    	long [][]matrix2 = new long[n2][m2];
    	for(int i = 0;i < n2;i++)
    		for(int j = 0;j < m2;j++)
    			matrix2[i][j] = sn.nextLong();
    			
    	long [][]result = new long[n1][m2];
    	for(int i = 0;i < n1;i++)
    		for(int j = 0;j < m2;j++)
    			for(int k = 0;k < m1;k++)
    			{
    				long temp = matrix1[i][k] * matrix2[k][j];
    				result[i][j] += temp;
    			}
    			
    	for(int i = 0;i < n1;i++) {
    			for(int j = 0;j < m2;j++)
	    			System.out.print(result[i][j]+" ");
    			System.out.println();
    	}
    	
    	sn.close();
    }
}

本人才疏学浅，请师傅们批评指教。