leetcode 73.矩阵置零

leetcode 73.矩阵置零

题干

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

进阶：
一个直观的解决方案是使用  O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗？
 
示例 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

题解

如果要原地修改，就不能遍历矩阵找到一个零处理一行一列，这样可能会覆盖同行/列存在的其他零，所以要每个零对行/列的影响。
开两个数组rows和cols分别统计矩阵中的0是否使得某行/列需置0,最后遍历两数组然后修改matrix即可。

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int n = matrix.size();
        if(n == 0) return;
        int m = matrix[0].size();
        vector<int> rows(n,0);
        vector<int> cols(m,0);
        for(int i = 0 ; i < n ; ++i){
            for(int j = 0 ; j < m ; ++j){
                if(matrix[i][j] == 0 && (!rows[i] || !cols[j])){
                    rows[i] = 1;
                    cols[j] = 1;
                }
            }
        }
        for(int i = 0 ; i < n ; ++i) 
            if(rows[i]) 
                for(int j = 0 ; j < m ; ++j) matrix[i][j] = 0;
        for(int i = 0 ; i < m ; ++i) 
            if(cols[i]) 
                for(int j = 0 ; j < n ; ++j) matrix[j][i] = 0;
        return;
    }
};

常数额外空间的解法本质上和数组记录行列数据是相同的，只是利用了原有矩阵的第一行和第一列来记录矩阵不同行列的置零状况，而多用两个变量来储存第一行/列的置零状况