刷题力扣面试题 01.08. 零矩阵

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zero-matrix-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

编写一种算法，若M × N矩阵中某个元素为0，则将其所在的行与列清零。

示例 1：

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

示例 2：

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

题目分析

1. 根据题目描述，将矩阵中0所在的行和列所有值置为0
2. 使用两个标记数组，记录记录矩阵中含有0的行和列
3. 在根据标记数组将矩阵中的行和列置0

代码

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int row = matrix.size();
        int col = matrix[0].size();
        std::vector<int> rowVec(row, 0);
        std::vector<int> colVec(col, 0);
        // 一次遍历，获取需要变动的行和列
        for (int i = 0; i < row; ++i) {
            for (int j = 0; j < col; ++j) {
                if (matrix[i][j] == 0) {
                    rowVec[i] = 1;
                    colVec[j] = 1;
                }
            }
        }
        // 将需要变动的行置为0
        for (int i = 0; i < row; ++i) {
            if (rowVec[i] == 1) {
                for (int j = 0; j < col; ++j) {
                    matrix[i][j] = 0;
                }
            }
        }
        // 将需要变动的列置为0
        for (int j = 0; j < col; ++j) {
            if (colVec[j] == 1) {
                for (int i = 0; i < row; ++i) {
                    matrix[i][j] = 0;
                }
            }
        }

    }
};