Leetcode73 矩阵置零

已AC, 中等题

思路:

  1. 设置两个集合,line_set,row_set分别存放矩阵中0所在的行和列,使用集合是避免重复遍历
  2. 第一次遍历二维数组,将相应数字装入集合
  3. 第二次遍历将两个集合里的行列全部置零

我的这种方法虽然通过,但是可能与题意不是很符合,题目要求原地算法,我额外占用了集合的空间应该是属于违规的,不借助额外空间也是可以的:

  • 每次遍历遇到0就把他所在的行列置位其他数字或字符比如 ‘ t ’,因为置为零会影响下面的判断,最后再遍历一次把 ‘ t ’字符的都置为零即可
class Solution(object):
    def setZeroes(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: void Do not return anything, modify matrix in-place instead.
        """
        i_set = set()
        j_set = set()
        for i in xrange(len(matrix)):
            for j in xrange(len(matrix[i])):
                if matrix[i][j] == 0:
                    i_set.add(i)
                    j_set.add(j)
        for i in i_set:
            matrix[i]=[0]*len(matrix[i])
        for j in j_set:
            for i in xrange(len(matrix)):
                matrix[i][j]=0
        #return matrix
        

 

点赞