73.矩阵置零

小小皮卡丘啃算法

已于 2024-04-06 13:49:39 修改

阅读量362

点赞数 2

分类专栏： leetcode热题100 文章标签：矩阵算法线性代数 leetcode python

于 2024-04-03 14:41:56 首次发布

本文链接：https://blog.csdn.net/weixin_57865902/article/details/137344288

版权

leetcode热题100 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

本文介绍了如何在二维矩阵中找到并置零的操作，首先通过暴力求解方法，记录0的位置；然后优化算法中，利用哈希表减少空间复杂度，将时间复杂度降低。

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

·题目描述

·解题思路

——————初步思路，暴力求解——————

1.遍历原矩阵，记录0位置的行列值，存入数组flags中，作为标志

2.读取flags中的元素，获取要置为0的行列值，然后两个循环将matrix中的对应元素置为0

代码：

    def setZeroes(self, matrix):
        #时间复杂度O（mn^2),空间复杂度o（mn）
        row = len(matrix)
        col = len(matrix[0])
        flags = []

        for i in range(row):
            for j in range(col):
                if matrix[i][j] == 0:
                    res = [i,j]
                    flags.append(res)

        for flag in flags:
            r,c = flag[0],flag[1]
            for i in range(row):
                matrix[i][c] =0
            for j in range(col):
                matrix[r][j] = 0

        return matrix

——————改进优化，使用哈希表——————

1.将原始需要记录行列的数组flags改为两个哈希表row和col

2.遍历矩阵将0位子的行列分别存入哈希表中

3.循环置为0

代码：

    def setZeroes(self, matrix):
          # 时间复杂度O(mn) 空间复杂度O(m+n)
        row_num = len(matrix)
        col_num = len(matrix[0]) if row_num > 0 else 0
        row = set()
        col = set()


        for i in range(row_num):
            for j in range(col_num):
                if matrix[i][j] == 0:
                    row.add(i)
                    col.add(j)

        for i in range(row_num):
            for j in range(col_num):
                if i in row or j in col:
                    matrix[i][j] = 0

        return matrix