题解 | #矩阵乘法#

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * @param a int整型vector<vector<>> 第一个矩阵
     * @param b int整型vector<vector<>> 第二个矩阵
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > solve(vector<vector<int> >& a, vector<vector<int> >& b) {
        int m = a.size();
        int p = a[0].size();
        int n = b[0].size();
        // 答案是m*n的矩阵
        vector<vector<int>> res(m, vector<int>(n));
        // 第i行
        for (int i = 0; i < m; ++i) {
            // 第j列
            for (int j = 0; j < n; ++j) {
                int sum = 0;
                // 应该等于a数组的第i行与b数组第j列的点积
                for (int k = 0; k < p; ++k) {
                    sum += a[i][k] * b[k][j];
                }
                res[i][j] = sum;
            }
        }
        return res;
    }
};