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]]

解题方法： 用两个一维数组保存对应的行或者对应列是否为0，先遍历数组中的元素matrix[i][j]，如果matrix[i][j]等于0，则将对应的行标记数组和列标记数组设置为true，表示对应的行或者列中的元素应该设置为0，再用这两个数组去更新原数组即可。（这里有一个优化的点，可以用两个变量分别保存第1行或第1列是否存在0，然后第1行和第1列的元素去取代两个一维数组的工作，最后用两个变量还原第1行和第一列的值）

C++代码

#include <iostream>
#include <vector>
class Solution {
public:
    void setZeroes(std::vector<std::vector<int>>& matrix) {
        int m = matrix.size(); // 行数
        int n = matrix[0].size(); // 列数
        std::vector<bool> rows_has_zero(m, false);//row_has_zero[i]表示第i+1行是否需要变为0
        std::vector<bool> cols_has_zero(n, false); // cols_has_zero[i]表示第i+1列是否需要变为0
        //第一步，生成标记数组
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (!matrix[i][j]) { //如果matrix[i][j]等于0，则将对应的行和列标记需要变为0
                    rows_has_zero[i] = true;
                    cols_has_zero[j] = true;
                }
            }
        }
        // 第二步，用标记数组去更新原数组
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (rows_has_zero[i] || cols_has_zero[j]) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
};