解题思路:
首先空间复杂度为O(m*n)的做法是,直接copy一套matrix来记录
空间复杂度为O(m+n)的做法是设置数组row[m]和column[n]来记录每行每列的0的情况
空间复杂度为O(1)的情况就是设置两个boolean变量,首先判定第一行和第一列是否会最终变为0,然后从第一行第一列(最左上角位置为第0行第0列)开始,更新matrix。最后再根据这两个boolean变量来更新matrix的第一行和第一列。
提交代码:
class Solution {
public void setZeroes(int[][] matrix) {
boolean zeroColumn=false,zeroRow=false;
for(int i=0;i<matrix.length;i++)
if(matrix[i][0]==0) {
zeroColumn=true;
break;
}
for(int j=0;j<matrix[0].length;j++)
if(matrix[0][j]==0) {
zeroRow=true;
break;
}
for(int i=0;i<matrix.length;i++)
for(int j=0;j<matrix[0].length;j++)
if(matrix[i][j]==0) {
matrix[i][0]=0;
matrix[0][j]=0;
}
for(int i=1;i<matrix.length;i++)
for(int j=1;j<matrix[0].length;j++)
if(matrix[i][0]==0||matrix[0][j]==0)
matrix[i][j]=0;
if(zeroColumn)
for(int i=0;i<matrix.length;i++)
matrix[i][0]=0;
if(zeroRow)
for(int j=0;j<matrix[0].length;j++)
matrix[0][j]=0;
}
}
运行结果: