题解 | #矩阵乘法#

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) {
        // write code here
        int n1 = a.size();
        int m1 = a[0].size();
        int n2 = b.size();
        int m2 = b[0].size();
        vector<vector<int>> res(n1, vector<int>(m2));
        if(m1 != n2)
            return res;
        for(int i = 0; i < n1; i++) {
            for(int j = 0; j < m2; j++) {
                for(int k = 0; k < m1; k++) {
                    res[i][j] += a[i][k] * b[k][j];
                }
            }
        }
        return res;
    }
};