LeetCode:73. Set Matrix Zeroes - Python

189 篇文章 3 订阅
151 篇文章 2 订阅

问题描述:

73. 矩阵置零

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

示例 :

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

进阶:

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

问题分析:

题目提示已经给出了,想必大家一定可以想到前两种方法,现在介绍另外一种方法,就不开辟空间的方法。

(1)要设置含有 0 的整列和整行全为 0,可以扫描一遍矩阵,在原始矩阵的第一列第一行,做标记,即,这一行或者这一列是否要置为 0

(2)最后根据第一列第一行的标志,对整列、行,进行置 0 处理。

(3)需要注意的是,要特别注意第一行和第一列含有 0 的情况,不能和标志混淆。

Python3实现:

class Solution:
    def setZeroes(self, matrix):
        
        first_col = False  # 用于记录 第一列是否有 0 没有为false

        n, m = len(matrix), len(matrix[0])

        for i in range(n):
            if matrix[i][0] == 0:  # 检测第一列是否有 0
                first_col = True
            for j in range(1, m):  # 从第二列开始,一行一行的扫描,然后在第一列和第一行上进行标记。
                if matrix[i][j] == 0:
                    matrix[0][j] = matrix[i][0] = 0

        for i in range(1, n):  # 根据第一列和第一行的标记,对整个矩阵置0处理,不包括第一行和第一列。
            for j in range(1, m):
                if matrix[i][0] == 0 or matrix[0][j] == 0:
                    matrix[i][j] = 0

        if matrix[0][0] == 0:  # 处理第一行
            for j in range(m):
                matrix[0][j] = 0

        if first_col:  # 处理第一列
            for i in range(n):
                matrix[i][0] = 0


if __name__ == '__main__':
    matrix = [
        [1, 1, 1],
        [1, 0, 1],
        [1, 1, 1]]
    solu = Solution()
    solu.setZeroes(matrix)
    for row in matrix:
        print(row)

声明: 总结学习,有问题或不妥之处,可以批评指正哦。

题目链接:leetcode-cn.com/problems/set-matrix-zeroes/ ,参考官方解答

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值