【leetcode面试经典150题】37. 矩阵置零（C++）

【题目描述】

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

【示例一】

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,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]]

【提示及数据范围】

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -2的31次方 <= matrix[i][j] <= 2的31次方 - 1

【代码】

// 使用标记数组

// 首先遍历该数组一次，如果某个元素为 0，
// 那么就将该元素所在的行和列所对应标记数组的位置置为 true。
// 最后我们再次遍历该数组，用标记数组更新原数组即可。

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