【题目】
Given a m x n matrix, if an element is 0,set its entire row and column to 0. Do it in place.
Follow up:
Did you use extra space?
A straight forward solution using O(mn)space is probably a bad idea.
A simple improvement uses O(m + n) space,but still not the best solution.
Could you devise a constant space solution?
输入一个m×n矩阵,如果出现有0,那么将对应的行和列都变成0. 要求空间复杂度为o(1)
【思路】
1、暴力法:设定一个同样大小的矩阵,每次遇到0,然后将第二个矩阵的相应位置置0,空间复杂度为O(mn)。
2、略微优化:每次遇到0的时候,记录需0的行号和列号,空间复杂度为O(m+n)。
3、空间复杂度为1,只需要将2中0的行号列号的记录在第一行和第一列就行了。
利用第一行和第一列的元素去标记该行或该列是否在更新时要全部变成0。但是这样操作会使得第一行和第一列的原始状态丢失。因此,我们需要额外一个变量hasZeros去保存第一列(或者第一行)在更新时是否要变成0,这样就不会有问题了。
另外还有一个让代码更加精简的小窍门,就是在更新的时候从下往上(Bottom-Up),这样就不用担心会破坏第一行或第一列的数据了。
【Python实现】
# -*- coding: utf-8 -*-
"""
Created on Sat Aug 5 14:44:15 2017
@author: Administrator
"""
class Solution(object):
def setZeroes(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: void Do not return anything, modify matrix in-place instead.
"""
m = len(matrix)
n = len(matrix[0])
if m == 0 or n == 0:
return
hasZeros = 0
for i in range(m):#第i行
if matrix[i][0] == 0:#第零列
hasZeros = 1
for j in range(1,n):#第i行的第j列
if matrix[i][j] == 0:
matrix[i][0] = 0
matrix[0][j] = 0
for i in range(m-1,-1,-1):#i = 2,1,0
for j in range(n-1,0,-1):#j = 2,1
if matrix[i][0] == 0 ormatrix[0][j] == 0:
#第零列的第i行,或第零行的第i列任一个为零,则将第i行第j列的元素置为零
matrix[i][j] = 0
if hasZeros == 1:
matrix[i][0] = 0
return
s = Solution()
matrix = [[1,0,2],[2,1,3],[2,1,1]]
s.setZeroes(matrix)