LC73. 矩阵置零

题目

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

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

示例 ：

输入：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]]

思路

题目中要求为常数空间复杂度，且用原地算法。
这里应记住：原地算法一般指取消标记数组，而在原数组内部进行标记
我们先按照常规思路进行思考：
为了避免判断一个数组元素 0 后将其所在行和列的所有元素都设为 0 ，导致原来数组出现的0被覆盖的现象，我们可以采用标记数组的方式。对于原数组中所有出现的0，标记其行号和列号。
最后统一将数组相应的行和列改为0。

我们通过常规思路知道了标记数组的作用，此时我们需要取消标记数组，采用原数组本身进行标记。
因为是要标记行和列，我们可以将原数组的第0行和第0列分别标记行和列。比如：当第j列需要设为0，则先将matrix[0][j]标为0.
但是这样有可能会把第0行和第0列原本的0覆盖掉。比如matrix[0][3]原本为0,标记第3列为0后，matrix[0][3]只代表第3列为0,但是我们还需要标记第0列为0.因此我们在标记开始前，先判断第0行和第0列中是否含有0，采用两个标记变量标记第0行和第0列是否需要后续改为0。
这样空间复杂度就变为常数级了

代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        bool f_r0 = false, f_l0 = false;
        for(int i=0;i<n;i++){
            if(matrix[0][i]==0)
                f_r0 = true;
        } 
        for(int i=0;i<m;i++){
            if(matrix[i][0]==0)
                f_l0 = true;
        }
        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;
                }
            }
        }
        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;
                }
            }
        }
        if(f_r0){
            for(int i=0;i<n;i++)
                matrix[0][i] = 0;
        }
        if(f_l0){
            for(int i=0;i<m;i++)
                matrix[i][0] = 0;
        }
    }
};