老卫带你学---leetcode刷题(73. 矩阵置零)

最新推荐文章于 2024-12-09 17:08:33 发布

老卫带你学

最新推荐文章于 2024-12-09 17:08:33 发布

阅读量423

点赞数 8

分类专栏： leetcode刷题文章标签： leetcode

本文链接：https://blog.csdn.net/yixieling4397/article/details/136354973

版权

leetcode刷题专栏收录该内容

120 篇文章

订阅专栏

73. 矩阵置零

问题

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

示例 1：
在这里插入图片描述

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]
示例 2：
在这里插入图片描述

输入：matrix = [[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（1））

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        row=len(matrix)
        col=len(matrix[0])
        row_zero=set()
        col_zero=set()
        for i in range(row):
            for j in range(col):
                if matrix[i][j]==0:
                    row_zero.add(i)
                    col_zero.add(j)
        for i in range(row):
            for j in range(col):
                if i in row_zero or j in col_zero:
                    matrix[i][j]=0