给定一个 m x n
的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
方法一、
我们可以用两个标记数组分别记录每一行和每一列是否有零出现。
具体地,我们首先遍历该数组一次,如果某个元素为 00,那么就将该元素所在的行和列所对应标记数组的位置置为true。最后我们再次遍历该数组,用标记数组更新原数组即可。
//法1 遍历两遍
public void setZeroes1(int[][] matrix){
int m=matrix.length;
int n=matrix[0].length;
boolean[] row=new boolean[m];
boolean[] col=new boolean[n];
for(int i=0;i<m;++i){
for(int j=0;j<n;++j){
if(matrix[i][j]==0){
row[i]=true;
col[j]=true;
}
}
}
for (int i=0;i<m;++i){
for(int j=0;j<n;++j){
if(row[i]||col[j]){
matrix[i][j]=0;
}
}
}
}
方法二、
我们可以用矩阵的第一行和第一列代替方法一中的两个标记数组,以达到 O(1)的额外空间。但这样会导致原数组的第一行和第一列被修改,无法记录它们是否原本包含 0。因此我们需要额外使用两个标记变量分别记录第一行和第一列是否原本包含 0。
public void setZeroes(int[][] matrix){
int m=matrix.length;
int n=matrix[0].length;
boolean rowFlag=false,columnFlag=false;
for(int i=0;i<n;++i){
if(matrix[0][i]==0){
rowFlag=true;
}
}
for(int j=0;j<m;++j){
if(matrix[j][0]==0){
columnFlag=true;
}
}
for (int i = 1; i < m; i++) {
for (int j = 1; j < n; j++) {
if (matrix[i][j] == 0) {
matrix[i][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(rowFlag){
for(int i=0;i<n;++i){
matrix[0][i]=0;
}
}
if(columnFlag){
for(int i=0;i<m;++i){
matrix[i][0]=0;
}
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/set-matrix-zeroes/solution/ju-zhen-zhi-ling-by-leetcode-solution-9ll7/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。