LeetCode题解(0073)：矩阵置零(Python)

最新推荐文章于 2024-09-23 11:27:55 发布

长行

最新推荐文章于 2024-09-23 11:27:55 发布

阅读量231

点赞数

分类专栏：刷题记录文章标签： leetcode python 数组

本文链接：https://blog.csdn.net/Changxing_J/article/details/112601220

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

该博客讨论了一种解决矩阵置零问题的优化算法，通过记录矩阵的首行和首列来减少不必要的操作，实现了时间复杂度为O(N×M)的空间复杂度为O(1)的解决方案。解法详细步骤包括统计首行首列0的位置，然后利用这些信息更新矩阵其余部分，最后更新首行和首列，从而达到目标效果。

摘要由CSDN通过智能技术生成

题目：原题链接（中等）

标签：数组

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O (N \times M)$	$O (1)$	48ms (71.66%)
Ans 2 (Python)
Ans 3 (Python)

解法一：

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        m, n = len(matrix), len(matrix[0])

        # 统计首行和首列是否存在0
        first_row, first_col = False, False
        for j in range(n):
            if matrix[0][j] == 0:
                first_row = True
                break
        for i in range(m):
            if matrix[i][0] == 0:
                first_col = True
                break

        # 统计其他位置，并将结果记录在首行和首列
        for i in range(1, m):
            for j in range(1, n):
                if matrix[i][j] == 0:
                    matrix[i][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 first_row:
            for j in range(n):
                matrix[0][j] = 0
        if first_col:
            for i in range(m):
                matrix[i][0] = 0