LeetCode73. Set Matrix Zeroes（思路及python解法）

最新推荐文章于 2024-10-08 15:23:51 发布

原创最新推荐文章于 2024-10-08 15:23:51 发布 · 322 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

179 篇文章

订阅专栏

Python

173 篇文章

订阅专栏

本文介绍了一种高效的算法，用于处理矩阵中若某一元素为0，则将其所在行和列的所有元素设为0的问题。通过使用numpy库的argwhere方法或原地标记策略，实现了在原矩阵上进行修改，避免了额外的空间开销。

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

Example 1:

Input: 
[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]
Output: 
[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

将0所在的行和列都重置为0。

第一想法比较简单，把所有0的所在行和所在列统计出来，然后将这些行和列设为0。当然这种方法时间复杂度很高。

这里学到np.argwhere()方法，能直接返回查找值所在行列。

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        import numpy as np

        rows=set()
        cals=set()
        m=np.matrix(matrix)
        for r,c in np.argwhere(m==0):
            rows.add(r)
            cals.add(c)
        for i in range(len(matrix)):
            for j in range(len(matrix[0])):
                if i in rows or j in cals:
                    matrix[i][j]=0
        return matrix

或者直接标记也很简单迅速。

就是在遍历过程中，把0所在的行和列都标记，最后再把所有标记的地方改为0。

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        rows = len(matrix)
        if(rows==0): return
        cols = len(matrix[0])
        for i in range(rows):
            for j in range(cols):
                if matrix[i][j] == 0:
                    for row in range(rows):
                        if matrix[row][j] != 0:
                            matrix[row][j] = 'x'
                    for col in range(cols):
                        if matrix[i][col] != 0: 
                            matrix[i][col] = 'x'

        for i in range(rows):
            for j in range(cols):
                if matrix[i][j] == 'x':
                    matrix[i][j] = 0