Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
这道题难点在于不能用额外空间,直接根据是否为零的状态改变原矩阵会覆盖没有遍历到的矩阵数值。如果用额外空间的话就先判断一遍用新的相同大小的矩阵存储状态,再根据那个矩阵改变出初始矩阵的值。不能用额外空间的话就考虑用第一行和第一列来保存这一行/列的状态。但是要首先用两个变量保存第一行和第一列的初始状态。
public void setZeroes(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
boolean firstrow = false;
boolean firstcol = false;
for(int i=0;i<n;i++){
//判断第一行初始
if(matrix[0][i]==0){
firstrow = true;
break;
}
}
for(int i=0;i<m;i++){
//判断第一列初始
if(matrix[i][0]==0){
firstcol = true;
break;
}
}
for(int i=1;i<m;i++){
//用第一行,第一列存储矩阵状态
for(int j=1;j<n;j++){
if(matrix[i][j]==0){
matrix[i][0]=0;
matrix[0][j]=0;
}
}
}
for(int i=1;i<m;i++){
//根据第一行第一列赋值
for(int j=1;j<n;j++){
if(matrix[i][0]==0||matrix[0][j]==0)
matrix[i][j]=0;
}
}
if(firstrow){
for(int i=0;i<n;i++)
matrix[0][i]=0;
}
if(firstcol){
for(int i=0;i<m;i++)
matrix[i][0]=0;
}
}