LeetCode | 矩阵置零

1
from typing import List
2

3

4
class Solution:
5
    def setZeroes(self, matrix: List[List[int]]) -> None:
6
        """
7
        Do not return anything, modify matrix in-place instead.
8
        """
9
        m = len(matrix)
10
        n = len(matrix[0])
11

12
        row_0 = [0] * m
13
        col_0 = [0] * n
14

15
        for row in range(0,m):
16
            for col in range (0,n):
17
                if matrix[row][col] == 0 :
18
                    row_0[row],col_0[col] = 1,1
19

20
        for row in range(0,m):
21
            for col in range (0,n):
22
                if (row_0[row] or col_0[col]) :
23
                    matrix[row][col] = 0

1
from src.python._073_Set_Matrix_Zeroes import Solution
2

3

4
sol = Solution()
5

6
matrix = [[1, 1, 1], [1, 0, 1], [1, 1, 1]]
7
assert sol.setZeroes(matrix) is None
8
assert matrix == [[1, 0, 1], [0, 0, 0], [1, 0, 1]]
9

10
matrix = [[0, 1, 2, 0], [3, 4, 5, 2], [1, 3, 1, 5]]
11
sol.setZeroes(matrix)
12
assert matrix == [[0, 0, 0, 0], [0, 4, 5, 0], [0, 3, 1, 0]]
13

14
matrix = [[1, 0]]
15
sol.setZeroes(matrix)
16
assert matrix == [[0, 0]]
17

18
matrix = [[1], [0]]
19
sol.setZeroes(matrix)
20
assert matrix == [[0], [0]]
21

22
print("PASS")