73. 矩阵置零

问题

给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

例子

思路

• 方法1

  空间复杂度O(m + n) ，boolean[] arr哪行有0【i<m】，哪列有0【i>=m, 行数为i-m】

• 方法2
  

• 方法3

可以用两个数，分别存储行和列是否有0【用二进制的一位表示是否为0】
代码

//方法1
class Solution {
    public void setZeroes(int[][] matrix) {
        int m=matrix.length, n=matrix[0].length;
        
        boolean[] arr = new boolean[m];
        boolean[] arr2= new boolean[n];
        for(int i=0; i<m; i++) {
            for(int j=0; j<n; j++) {
                if(matrix[i][j]==0) {
                    arr[i]=true;
                    arr2[j]=true;
                }
            }
        }
        for(int i=0; i<arr.length; i++) {
            if(arr[i]==true) 
                for(int j=0; j<n; j++) matrix[i][j]=0;
   
        }
        for(int i=0; i<arr2.length; i++) {
            if(arr2[i]==true) 
                for(int j=0; j<m; j++) matrix[j][i]=0;

        }
        
        return;

    }
}
//方法2
class Solution {
    public void setZeroes(int[][] matrix) {
        int m=matrix.length, n=matrix[0].length;
        boolean row_0=false,col_0=false;
        //第一行是否有0
        for(int i=0; i<m; i++) 
            if(matrix[i][0]==0) col_0=true;
        //第一列是否有0
        for(int j=0; j<n; j++)
            if(matrix[0][j]==0) row_0=true;
        //把元素的所在的行首和列首置为0
        //最开始时首行首列是否有零，此时可能会被覆盖掉，所以用row_0,col_0存储首行首列是否有0
        for(int i=0; i<m; i++) {
            for(int j=0; j<n; j++) {
                if(matrix[i][j]==0) {
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }
        //遍历非首行首列的元素，如果该元素的行首或列首为0，则该元素为0
        for(int i=1; i<m; i++) {
            for(int j=1; j<n; j++) {
                if(matrix[0][j]==0 || matrix[i][0]==0)
                    matrix[i][j]=0;
            }
        }
        //最开始时，首行和首列是否有0，若有0，将首行或首列变为0
        if(row_0){
            for(int j=0; j<n; j++)
                matrix[0][j]=0;
        }
        if(col_0) {
            for(int i=0; i<m; i++)
                matrix[i][0]=0;
            
        }
        return;

    }
}