FilesExpand file tree

 master

/

sparse-matrix-multiplication.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

18 lines (17 loc) · 562 Bytes

 master

/

sparse-matrix-multiplication.py

Copy path

Top

File metadata and controls

• Code
• Blame

18 lines (17 loc) · 562 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

# Time: O(m * n * l), A is m x n matrix, B is n x l matrix

# Space: O(m * l)

class Solution(object):

def multiply(self, A, B):

"""

:type A: List[List[int]]

:type B: List[List[int]]

:rtype: List[List[int]]

"""

m, n, l = len(A), len(A[0]), len(B[0])

res = [[0 for _ in xrange(l)] for _ in xrange(m)]

for i in xrange(m):

for k in xrange(n):

if A[i][k]:

for j in xrange(l):

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

return res