leetcode_73_矩阵置零
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
class Solution {
public void setZeroes(int[][] matrix) {
//新思路,使用矩阵的第一行和第一列来存储矩阵该行和该列的某些重要特性
//在本题中,使用第一行和第一列来记载该列是否有元素为零,而第一列是否有零使用一个布尔变量来记载
int m = matrix.length;
int n = matrix[0].length;
boolean flagcol1 = false;
for(int i=0;i<m;i++){
if(matrix[i][0]==0){
flagcol1 = true;
}
for(int j=1;j<n;j++){
if(matrix[i][j]==0){
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
//因为第一行是标记位,我们为了不提前更改标记位的值,需要从最后一行开始遍历,而第一列因为使用了额外的
//布尔变量来存储该列是否有零,因此不用遍历,只需要检测完该行后根据需要将第一列的相应行变零即可
for(int i=m-1;i>=0;i--){
for(int j=1;j<n;j++){
if(matrix[i][0]==0||matrix[0][j]==0){
matrix[i][j]=0;
}
}
if(flagcol1){
matrix[i][0]=0;
}
}
}
}