LeetCode--Set matrix zeros

题目:

给定一个数组matrix(m*n),若某个位置值为0,则将该行该列的值都设为0

思路:

最直接的想法是,直接对数组进行遍历,遇到0则将该行该列都置0.但这样会导致重复和混乱。

再想到重新建立一个矩阵,新矩阵的值来记录该位置是原值还是0,但比较浪费空间O(m*n)。

考虑空间后,新建立一个O(m+n)的向量,用来存储对应行和列是否置零O(m+n)。

再考虑用首行和首列来存储对应行和列是否置零。而首行首列是否置零存储在外部连个变量里O(1).


代码(python):

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值