0311：稀疏矩阵的乘法（★）

力扣第 311 题

题目

给定两个 稀疏矩阵 ：大小为 m x k 的稀疏矩阵 mat1 和大小为 k x n 的稀疏矩阵 mat2 ，返回 mat1 x mat2 的结果。你可以假设乘法总是可能的。

示例 1：

输入：mat1 = [[1,0,0],[-1,0,3]], mat2 = [[7,0,0],[0,0,0],[0,0,1]]
输出：[[7,0,0],[-7,0,3]]

示例 2:

输入：mat1 = [[0]], mat2 = [[0]]
输出：[[0]]

提示:

• m == mat1.length
• k == mat1[i].length == mat2.length
• n == mat2[i].length
• 1 <= m, n, k <= 100
• -100 <= mat1[i][j], mat2[i][j] <= 100

分析

暴力计算即可。

解答

1
2

def multiply(self, mat1: List[List[int]], mat2: List[List[int]]) -> List[List[int]]:
    return [[sum(a*b for a, b in zip(row, col)) for col in zip(*mat2)] for row in mat1]

96 ms