矩阵置零

发表于 | 分类于 |
字数统计: 355 | 阅读时长 ≈ 1

矩阵置零

给定一个 $m x n$ 的矩阵，如果一个元素为 0 ，则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

进阶：

  • 一个直观的解决方案是使用 $O(mn)$ 的额外空间，但这并不是一个好的解决方案。
  • 一个简单的改进方案是使用 $O(m + n)$ 的额外空间，但这仍然不是最好的解决方案。
  • 你能想出一个仅使用常量空间的解决方案吗？

思路：

先用第一行，第一列来标记对应的行，列中是否出现了0。 因为第一行，第一列会被修改，所以先用两个flag来记录第一行，第一列中是否有0。最后在处理第一行，第一列。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        n, m = len(matrix), len(matrix[0])
        row_flag, col_flag = False, False
        for j in range(m):
            if matrix[0][j] == 0:
                row_flag = True
                break

        for i in range(n):
            if matrix[i][0] == 0:
                col_flag = True
                break
            
        for i in range(n):
            for j in range(m):
                if matrix[i][j] == 0:
                    matrix[i][0] = 0
                    matrix[0][j] = 0
        
        for i in range(1, n):
            if matrix[i][0] == 0:
                for j in range(1, m):
                    matrix[i][j] = 0
        
        for j in range(1, m):
            if matrix[0][j] == 0:
                for i in range(1, n):
                    matrix[i][j] = 0
        
        if row_flag == True:
            for j in range(m):
                matrix[0][j] = 0
        if col_flag == True:
            for i in range(n):
                matrix[i][0] = 0

0%