题目:Submission Details
难度:medium
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
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?
解题思路:给一个矩阵,如果matrix[I][j]=0,则让该行该列的数全置0.
看到此题,第一想法是遍历矩阵,如果遇到0,则行列置0.这是错误的解法!比如:
一个矩阵0 0 0 5
4 3 1 4
0 1 1 4
1 2 1 3
0 0 1 1
遇到第一个0,将b[1][0]变成了0,等遍历到b[1][0],b[1][0]的值已经变了,解题也会出错。
第二个想法是构建一个平行矩阵matrix2[I][j],遍历matrix[I][j],然后将变化填入matrix2[I][j]中即可。但是使用了额外空间,占用了额外O(n*m)内存。
接着我想到了之前看到的N皇后问题的优化解,可以使用两个数组,hang[row]和lie[column]来表示矩阵的第row行,第column是否被置0,这样就只用了O(m+n),简便了许多。
具体代码如下:
public static void setZeroes(int[][] matrix) {
int row=matrix.length;
int column=matrix[0].length;
boolean[] hang=new boolean[row];
boolean[] lie=new boolean[column];
for(int i=0;i<row;i++){
for(int j=0;j<column;j++){
if(matrix[i][j]==0){
hang[i]=true;
lie[j]=true;
}
}
}
for(int i=0;i<row;i++){
if(hang[i]){
for(int j=0;j<column;j++){
matrix[i][j]=0;
}
}
}
for(int i=0;i<column;i++){
if(lie[i]){
for(int j=0;j<row;j++){
matrix[j][i]=0;
}
}
}
}