思路:

题目的主要信息:

  • 求两个矩阵相乘结果,两个矩阵一定存在,不用排除特殊情况
  • 两个矩阵都为的矩阵

方法一:暴力法
具体做法:
直接遍历二维矩阵的每个元素,其中等于a b两个矩阵的行乘列叠加即可。

class Solution {
public:
    vector<vector<int> > solve(vector<vector<int> >& a, vector<vector<int> >& b) {
        int n = a.size(); 
        vector<vector<int> > res(n, vector<int>(n, 0)); //构建结果矩阵
        for(int i = 0; i < n; i++)
            for(int j = 0; j < n; j++)
                for(int k = 0; k < n; k++)
                    res[i][j] += a[i][k] * b[k][j]; //行*列相加
         return res;
    }
};

复杂度分析:

  • 时间复杂度:,三层循环
  • 空间复杂度:,res是必要空间,不属于额外空间

方法二:暴力法缓存优化
具体做法:
调换代码顺序,依据cpu连续读取的特性,优先访问a[i][k],利用更多的空间连续性,虽然时间复杂度没有变,但是可以节省更多时间。
图片说明

class Solution {
public:
    vector<vector<int> > solve(vector<vector<int> >& a, vector<vector<int> >& b) {
        int n = a.size();
        int temp;
        vector<vector<int> > res(n, vector<int>(n, 0)); //构建结果矩阵
        for(int i = 0; i < n; i++)
            for(int k = 0; k < n; k++){
                temp = a[i][k];  //优先访问这个
                for(int j = 0; j < n; j++)
                    res[i][j] += temp * b[k][j]; //行*列相加
            }
         return res;
    }
};

复杂度分析:

  • 时间复杂度:,三层循环
  • 空间复杂度:,res是必要空间,不属于额外空间