今天的每日一题太难了 做了道矩阵置零

矩阵元素置零的处理方法
class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        m = len(matrix)
        n = len(matrix[0])
        seed = []
        hang_jihe = []
        for i in matrix:
            hang_jihe.append(set(i))
        lie_jihe = [set() for i in range(n)]#列

        for  i in range(m):#行
            for j in range(n):
                lie_jihe[j].add(matrix[i][j])
        


        
        for i in range(m):
            for j in range(n):
                if 0 in hang_jihe[i] or 0 in lie_jihe[j]:
                    matrix[i][j] = False
        for i in range(m):
            for j in range(n):
                if matrix[i][j] == False:
                    matrix[i][j] = 0
        
        return matrix
                    

将每一行的元素加入行集合中,将每一列的元素加入列集合中,在遍历到元素所在行集合或者列集合中有0时,将其改为False,再遍历一次,将False改为0 ,就绕开非初始化0 的重复判断。

粘贴一段大佬写的

class Solution:
    def setZeroes(self, matrix: List[List[int]]) -> None:
        """
        Do not return anything, modify matrix in-place instead.
        """
        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

他只在集合中记录了应该变成集合的行和应该变成集合的列,在遍历到相应行列时就改元素为0,佩服。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值