【C++学习笔记】数组与字符串——零矩阵

题目：零矩阵

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

题目来源：

   https://leetcode.cn/leetbook/read/array-and-string/ciekh/

题目内容：

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

C++代码实现

建立两个数组arr_row和arr_col，分别记录需要清零的行和列，数组元素的值=1时，表示该行或该列需要清零。

时间复杂度：O(MN)，空间复杂度：O(M+N)

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int M = matrix.size(), N = matrix[0].size();
        if(M == 0 || N == 0) return;
        vector<int> arr_row(M)