LeetCode[python] 矩阵置零

最新推荐文章于 2025-01-10 09:30:28 发布

原创最新推荐文章于 2025-01-10 09:30:28 发布 · 704 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

9 篇文章

订阅专栏

博客围绕一个 m x n 矩阵元素置零问题展开，若元素为 0，则将其所在行和列元素都设为 0，要求使用原地算法。提到直接方案用 O(mn) 额外空间不佳，改进方案用 O(m + n) 额外空间也非最优，最后给出 O(m + n) 额外空间的解决思路及代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例1：


输入: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
输出: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

示例2：

输入: 
[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]
输出: 
[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

进阶:

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

思路：

这里给出一个O(m+n)的额外空间方法来解决。即分别定义两个list : zeroLine 和 zeroRow来储存出现零的位置。

代码：

class Solution:
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: void Do not return anything, modify matrix in-place instead.
        """
        lenRow = len(matrix[0])
        lenLine = len(matrix)

        zeroRow = [False] * lenRow
        zeroLine = [False] * lenLine

        for l in range(lenLine):
            for r in range(lenRow):
                if matrix[l][r] == 0:
                    zeroRow[r] = True
                    zeroLine[l] = True

        for l in range(lenLine):
            if zeroLine[l] is True:
                matrix[l] = [0] * lenRow
            else:
                for r in range(lenRow):
                    if zeroRow[r] is True:
                        matrix[l][r] = 0

        return matrix