Leetcode 311 / Lintcode 654 Sparse Matrix Multiplication

• LQ 要钱故跑到 Lintcode去做…
• 首先看暴力解:
  • • 就是res[i][j] 等于A的第i行和B第j列的点积

class Solution {
public:
    vector<vector<int>> multiply(vector<vector<int>> &A, vector<vector<int>> &B) {
        int m = A.size(), n = B[0].size(), t = A[0].size();
        vector<vector<int>> res(m, vector<int>(n, 0));

        for(int i = 0; i < m; i ++)
            for(int j = 0; j < n; j ++)
                for(int k = 0; k < t; k ++)
                    res[i][j] += A[i][k] * B[k][j];

        return res;
    }
};

• 注意到 其实这三重循环的先后顺序，和最终求解是无关的
• 且，考虑到稀疏性，很多计算是没有必要的。关键是能直接提取出A或B矩阵中某个元素，从而可以判断是否为0，然后跳过一些计算
• 因此直接调换顺序，再加上判断，如下：
• 实际意义上理解就是：找到A中的非零元素A[i][k]，用它和对应需要计算的元素计算。
  • 此时i，k固定，j移动
  • 具体就是和[i][k] 要乘的对应第k行中的元素们B[k][j] 相乘

class Solution {
public:
    vector<vector<int>> multiply(vector<vector<int>> &A, vector<vector<int>> &B) {
        int m = A.size(), n = B[0].size(), t = A[0].size();
        vector<vector<int>> res(m, vector<int>(n, 0));

        for(int i = 0; i < m; i ++)
            for(int k = 0; k < t; k ++){
                if(!A[i][k]) continue;
                for(int j = 0; j < n; j ++)
                    res[i][j] += A[i][k] * B[k][j];
            }

        return res;
    }
};