NC143 矩阵乘法 JAVA

Posted 2022-12-07 蒙面侠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;