Leetcode 73 Python

lovefreedom22

于 2018-02-07 10:36:36 发布

阅读量432

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.csdn.net/u010429989/article/details/79277448

Leetcode 专栏收录该内容

48 篇文章

订阅专栏

本文介绍了一种在常数空间复杂度下将包含0元素的矩阵对应行和列置零的算法实现。通过巧妙利用矩阵的第一行和第一列作为标记位，并额外记录这两列是否需要置零，最终达到在原地修改矩阵的目的。

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

解题思路：需使用constant-space，因此不能根据matrix大小额外开辟空间。所以使用第0行，第0列来记录。并记录其中是否有0，若有0则最后将0行0列全部置0。

class Solution(object):
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: void Do not return anything, modify matrix in-place instead.
        """
        if matrix == []:
            return
        column = False
        row = False
        m = len(matrix)
        n = len(matrix[0])
        for i in range(m):
            if matrix[i][0] == 0:
                row = True
                break
        for j in range(n):
            if matrix[0][j] == 0:
                column = True
                break
        for i in range(1,m):
            for j in range(1,n):
                if matrix[i][j] == 0:
                    matrix[i][0] = 0
                    matrix[0][j] = 0 
        for i in range(1,m):
            if matrix[i][0] == 0:
                for j in range(1,n):
                    matrix[i][j] = 0
        for j in range(1,n):
            if matrix[0][j] == 0:
                for i in range(1,m):
                    matrix[i][j] = 0
        if column:
            for j in range(n):
                matrix[0][j] = 0
        if row:
            for i in range(m):
                matrix[i][0] = 0