Leetcode 311. Sparse Matrix Multiplication (python)

努利！奋斗！

已于 2022-06-17 12:00:20 修改

阅读量1.1k

点赞数

分类专栏： Leetcode 其他类型题目文章标签： python leetcode

于 2020-09-27 09:30:50 首次发布

本文链接：https://blog.csdn.net/qq_37821701/article/details/108821029

版权

这篇博客介绍了LeetCode 311题——稀疏矩阵乘法的解法。首先，讲解了暴力求解矩阵乘法的基本思路，然后提出了使用HashMap优化的解法，将稀疏矩阵转换为稠密矩阵进行乘法运算。在第二次刷题中，博主使用C++实现，并解决了C++中不能直接使用pair或vector作为key的问题，且优化后的解法无需解码过程，时间复杂度降低。

摘要由CSDN通过智能技术生成

Leetcode 311. Sparse Matrix Multiplication

题目
解法：暴力
解法2：
二刷

题目

在这里插入图片描述

解法：暴力

这边主要讲一下怎么做矩阵乘法。假设我们有矩阵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