73.矩阵置零
这道题相对简单。
首先我们需要标记需要置零的行和列,可以在遍历矩阵中的元素遇到0,则将其行首和列首元素置为0。在此过程中首行、首列会受影响,因此先用两个变量记录首行、首列是否需要被置0,接着遍历非首行、非首列的元素。
class Solution {
public void setZeroes(int[][] matrix) {
int n = matrix.length;
int m = matrix[0].length;
boolean c0=false, r0=false;
for(int i=0; i<n; i++){
if(matrix[i][0]==0){
c0=true;
break;
}
}
for(int j=0; j<m; j++){
if(matrix[0][j]==0){
r0=true;
break;
}
}
for(int i=1; i<n; i++){
for(int j=1; j<m; j++){
if(matrix[i][j]==0){
matrix[i][0]=matrix[0][j]=0;
}
}
}
for(int i=1; i<n; i++){
if(matrix[i][0]==0){
Arrays.fill(matrix[i],0);
}
}
for(int j=1; j<m; j++){
if(matrix[0][j]==0){
for(int i=1; i<n; i++){
matrix[i][j]=0;
}
}
}
if(r0){
Arrays.fill(matrix[0],0);
}
if(c0){
for(int i=0; i<n; i++){
matrix[i][0]=0;
}
}
}
}