73. 矩阵置零【 力扣(LeetCode) 】

一、题目描述

给定一个 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

三、解题思路

1. 基本思路：
     一开始最好别直接去考虑原地算法，先考虑最暴力的解法，新开一个矩阵，则时候我们只需要遍历原来的矩阵，遇到 0 ，就在新矩阵中将对应行和列置 0 ；
     这时刻可以发现，很多情况下，置 0 都是重复操作，同一行或者同一列只要有 1 个 0 就应该置 0 ，有什么办法可以让置 0 操作一行或者一列只做一次呢？那就要事先确定该行或列要置 0 ，可以使用标记向量，其中元素为 0 表示该行或列要置 0 ，这时候空间就从 $\mathrm{O}(mn)$ 优化到 $\mathrm{O}(m+n)$ ；
     在考虑原地算法，有什么办法可以再省去这 $\mathrm{O}(m+n)$ 的空间呢？那只能把标记向量放到矩阵中，可以让第一行和第一列代替标记向量。【其实第一行可以看出是遍历完第一行后的标记向量，同理，第一列也是】不过要考虑第一行第一个元素，它为 0 可能是第一行元素为 0 ，也可能是第一列元素为 0 ，这无法确定，所以申请两个变量，单独储存第一行和第一列是否要置 0 。
2. 具体思路：
   • 打标记：遍历矩阵，第 i 行第 j 个元素为 0 ，则第 1 行第 j 个元素标记为 0 ，第 1 列第 i 个元素标记为 0 。如果 i==0 ，则 r1=0 ；如果 j==0，则 c1=0 ；
   • 置 0 ：
     • 除去第 1 行外，根据第 1 列来判断，其他行该置 0 则置 0
     • 除去第 1 列外，根据第 1 行来判断，其他列该置 0 则置 0
     • 根据 r1 和 c1 判断，第一行和第一列该置 0 的置 0

四、参考代码

时间复杂度：$\mathrm{O}(mn)$
空间复杂度：$\mathrm{O}(1)$

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int n = matrix.size(), m = matrix[0].size();
        int r1 = 1, c1 = 1;

        for (int i = 0; i < n; i++) { // 打标记
            for (int j = 0; j < m; j++) {
                if (matrix[i][j] == 0) {
                    if (i == 0) {
                        r1 = 0;
                    }
                    if (j == 0) {
                        c1 = 0;
                    }
                    matrix[i][0] = matrix[0][j] = 0;
                }
            }
        }

        for (int i = 1; i < n; i++) { // 除去第 1 行外，其他行该置 0 则置 0
            if (matrix[i][0] == 0) {
                for (int j = 1; j < m; j++) {
                    matrix[i][j] = 0;
                }
            }
        }
        for (int j = 1; j < m; j++) { // 除去第 1 列外，其他列该置 0 则置 0
            if (matrix[0][j] == 0) {
                for (int i = 1; i < n; i++) {
                    matrix[i][j] = 0;
                }
            }
        }
        if (r1 == 0) { // 判断第 1 行是否置 0
            for (int j = 0; j < m; j++) {
                matrix[0][j] = 0;
            }
        }
        if (c1 == 0) { // 判断第 1 列是否置 0
            for (int i = 0; i < n; i++) {
                matrix[i][0] = 0;
            }
        }
    }
};