矩阵练习1

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;}}}
}