解题思路:这道题很容易上手,是典型的矩阵操作问题,但是其中的陷阱是,我们遇到0的时候不能直接把矩阵的行列在当前矩阵直接置0,否则后面还没访问到的会被当成原来是0,最后会把很多不该置0的行列都置0了。我们看到其实判断某一项是不是0只要看它对应的行或者列应不应该置0就可以,所以我们可以维护一个行和列的判断数组,然后扫描一遍矩阵记录那一行或者列是不是应该置0即可,后面赋值是一个常量时间的判断。这种方法的空间复杂度是O(m+n)。
代码如下:
class Solution {
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
if(m==0 || n==0)
return;
int[] flagr = new int[m];
int[] flagc = new int[n];
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[i][j]==0){
flagr[i] = 1;
flagc[j] = 1;
}
}
}
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(flagr[i]==1 || flagc[j]==1)
matrix[i][j] = 0;
}
}
}
}