矩阵乘法。

题目描述

描述
给定两个 nnn∗n 的矩阵 AA 和 BB ，求 ABA∗B 。

数据范围：
1 \le n \le 5001≤n≤500
-100 \le n \le 100−100≤n≤100
示例1
输入：
[[1,2],[3,2]],[[3,4],[2,1]]

返回值：
[[7,6],[13,14]]

代码

import java.util.*;
public class Solution {
    public int[][] solve (int[][] a, int[][] b) {
        // write code here
            int m = a.length;//A的行数
            int p = a[0].length;//A的列数=B的行数
            int n = b[0].length;//B的行数
            int[][] ans = new int[m][n];
            for(int i = 0 ; i < m ; i++)
            {
                for(int j = 0 ; j < n ; j++)
                {
                        int t = 0;
                        for(int k = 0 ; k < p ; k++)
                        {
                                t+=a[i][k]*b[k][j];
                        }
                        ans[i][j]=t;
                }
            }
            return ans;
    }
}