73.矩阵置零

1.题目描述

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

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]

示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]

提示：

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -231 <= matrix[i][j] <= 231 - 1

进阶：

• 一个直观的解决方案是使用  O(mn) 的额外空间，但这并不是一个好的解决方案。
• 一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
• 你能想出一个仅使用常量空间的解决方案吗？

2.解题思路

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

使用两个数组，rows用于记录该行是否包含0，cols用于记录该列是否包含0，一次遍历后，rows,cols中分别保存了所有含0的行和列的情况，然后再遍历矩阵，给rows[i] = true 或 cols[j] = true的位置赋0即可，这种方式使用了O(m+n)的空间复杂度，可以进一步优化

2.2空间复杂度O(1)

上述思路是借助两个数组统计的某一行某一列的含0情况，而为了不使用额外的空间，我们可以就在matrix数组的第0列，第0行的位置记录 从[1,m-1)行中为0的情况，和[1,n-1)列中为0的情况，就相当于把rows,cols两个数组转化成了matrix的第0列和第0行，但是这么做会使得原来第0行第0列的情况没办法被记录下来，因此就需要两个flag标记位，先对第0行第0列是否含0进行记录，然后其他的位置 如 matrix[i][j]如果为0，就将matrix[0][j] 和 matrix[i][0]置为0，后面遍历遇到这个位置为0时，要把相应行和列的位置上也都置为0

3.代码实现

3.1方法一

/**
     * 基于两个辅助数组rows cols,空间复杂度为O(m+n)
     * @param matrix
     */
    public void setZeroes(int[][] matrix) {
        //统计哪一行 哪一列需要置为0
        int m = matrix.length;
        int n = matrix[0].length;
        boolean[] rows = new boolean[m];
        boolean[] cols = new boolean[n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 0) {
                    rows[i] = true;
                    cols[j] = true;
                }
            }
        }
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (rows[i] || cols[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }

3.2方法二

class Solution {
    public void setZeroes(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        //flag1,flag2分别用于统计第0行，第0列是否含有0
        boolean flag1 = false;
        boolean flag2 = false;
        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) {
                flag1 = true;
            }
        }
        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                flag2 = true;
            }
        }
        
        //将第0行 和 第0列当作方法一中的rows 和 cols 辅助数组
        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
        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;
                }
            }
        }
        //根据flag1，flag2的值，分别给第0行第0列赋0
        if (flag1) {
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }
        if (flag2) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
    }
}