Leetcode 311. Sparse Matrix Multiplication (python)

题目

解法：暴力

这边主要讲一下怎么做矩阵乘法。假设我们有矩阵A，B，结果是C，那么

C[i][k] += A[i][j]*B[j][k]

从矩阵乘法的维度变化来考虑就可以理解了。
遍历的顺序不重要，下面几种都是ok的

class Solution:
    def multiply(self, A: List[List[int]], B: List[List[int]]) -> List[List[int]]:
        ans = [[0]*len(B[0]) for _ in range(len(A))]
        
        # for i in range(len(A)):
        #     for j in range(len(A[0])):
        #         for b_j in range(len(B[0])):
        #             ans[i][b_j] += A[i][j]*B[j][b_j]
        
        for i in range(len(A)):
            for b_i in range(len(B)):
                for b_j in range(len(B[0])):
                    ans[i][b_j] += A[i][b_i]*B[b_i][b_j]
        
        return ans

解法2：

利用hashmap来储存非0元素。这应该是面试中比较标准的解法。一个encode来把sparse_matrx转换成dense_matrix。然后对