NC143 矩阵乘法 JAVA

Posted 2021-08-17 蒙面侠1024

描述
给定两个n* n的矩阵A和B，求A*B。

套用矩阵算法模板

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 
     * @param a int整型二维数组 第一个矩阵
     * @param b int整型二维数组 第二个矩阵
     * @return int整型二维数组
     */
    public int[][] solve (int[][] a, int[][] b) {
        // write code here
        int n=a.length;
        int[][] res=new int[n][n];
		for(int i=0;i<n;i++) {
			for(int j=0;j<n;j++) {
				int sum=0;
				for(int k=0;k<n;k++) {
					sum+=a[i][k]*b[k][j];
				}
				res[i][j]=sum;
			}
		}
		return res;
    }
}