Leetcode hot100 矩阵置零【中等】

法（一）一个非常简单，3分钟就能写出来的解法

先遍历一遍矩阵，用boolean[] row[] 和boolean[] col去记录哪些行和哪些列是0.想·

然后再遍历一遍矩阵，置0

java 复制代码

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;  //行
        int n = matrix[0].length;  //列

        boolean[] row = new boolean[m];
        boolean[] col = new boolean[n];
        
        //标记哪些行，哪些列是0
        for(int i=0;i<m;i++){
            for (int j=0; j<n; j++){
                if(matrix[i][j]==0){
                    row[i]=true;
                    col[j]=true;
                }
            }
        }
        
        //填充0
        for(int i=0;i<m;i++){
            for (int j=0; j<n; j++){
                if(row[i]||col[j]){
                    matrix[i][j]=0;
                }
            }
        }
    }
}

空间复杂度：O(m+n)

然后，让用O(1)的空间复杂度，给出的方法是，占用原矩阵第一行和第一列用来标记。。。无语了，为了省而省，一点也不优雅。。

但是呢，也有很多要考虑的地方。

（1）考虑既然在原有矩阵基础上标记，那就不能用boolean类型了，只能用int类型，如何标记呢？分析了一下，用0标记就好了。这样最后我们填充0的时候，只要matrix[i][0]==0（或者matrix[0][i]==0），这一行（列）就都标记0。即使是matrix[i][0]本来就是0，那最后填充的时候，这一行也都标记为0，没毛病。

（2）考虑需要一开始对第一行和第一列做清空处理吗，还是什么都不用做。答案是什么都不用做。保留第一行和第一列的原数据，然后遍历去掉第一行和第一列的"小矩形"，去覆盖第一行和第一列的值。

（3）还有就是，不能只管"小矩形"啊。一开始要遍历第一行和第一列，看看第一行和第一列里是否有0，这里会用到额外的空间去记录

java 复制代码

class Solution {
    public void setZeroes(int[][] matrix) {
                int m = matrix.length;  //行
        int n = matrix[0].length;  //列

        boolean firstRow = false;
        boolean firstCol = false;
        
        //记录第一行和第一列是否有0
        for(int j=0; j<n; j++){
            if(matrix[0][j]==0) firstRow=true;
        }
        
        for (int i=0; i<m; i++){
            if(matrix[i][0]==0) firstCol=true;
        }
        
        //标记哪些行，哪些列是0
        //注意，遍历的是小矩阵，从i和j都是1开始
        for(int i=1;i<m;i++){
            for (int j=1; j<n; j++){
                if(matrix[i][j]==0){
                    matrix[i][0]=0;
                    matrix[0][j]=0;
                }
            }
        }

        //填充0
        //注意，填充的是小矩阵，从i和j都是1开始
        for(int i=1;i<m;i++){
            for (int j=1; j<n; j++){
                if(matrix[i][0]==0||matrix[0][j]==0){
                    matrix[i][j]=0;
                }
            }
        }
        
        //最后填充第0行和第0列
        if(firstRow){
            for(int j=0; j<n; j++){
               matrix[0][j]=0;
            }
        }
        
        if(firstCol){
            for (int i=0; i<m; i++){
                matrix[i][0]=0;
            }
        }
    }
}

整个过程有点像穿脱原则。