矩阵练习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;
            }
        }
    }
}