已AC, 中等题
思路:
- 设置两个集合,line_set,row_set分别存放矩阵中0所在的行和列,使用集合是避免重复遍历
- 第一次遍历二维数组,将相应数字装入集合
- 第二次遍历将两个集合里的行列全部置零
我的这种方法虽然通过,但是可能与题意不是很符合,题目要求原地算法,我额外占用了集合的空间应该是属于违规的,不借助额外空间也是可以的:
- 每次遍历遇到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