编写一种算法,若M × N矩阵中某个元素为0,则将其所在的行与列清零。
示例 1:
输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
示例 2:
输入:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
输出:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
每次找到等于零的那一个把对应的行和列都置零即可,但是会出现一个问题就是到最后所有的元素都变成零了;
因此,既然打算将整行和整列清零,因此并不需要标记录它是cell[2][4](行2,列4),只需要知道行2有个元素为0,列4有个元素为0.不管怎样,整行和整列都要清零,又何必要记录零元素的确切位置?
class Solution {
public void setZeroes(int[][] matrix) {
int M=matrix.length; //M为二维数组的行向量
int N=matrix[0].length; //N为二维数组的列向量
int[] rows =new int[M]; //新建两个数组
int[] cols =new int[N];
Arrays.fill(rows,0); //将数组中的元素全部填充为0
Arrays.fill(cols,0);
//先找到为0的元素的位置,
for(int i=0;i<M;i++){
for(int j=0;j<N;j++){
if(matrix[i][j]==0){
rows[i]=1;
cols[j]=1;
}
}
}
遍历两个一维数组,判断如果元素为1,将行列赋值为0
for(int i=0;i<M;i++){
for(int j=0;j<N;j++){
if(rows[i]==1 || cols[j]==1)
matrix[i][j]=0;
}
}
}
}