力扣：零矩阵

编写一种算法，若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]
]

思路：
标记零的位置，对零的位置的该行该列进行值的修改，修改为零。

代码：

class Solution {
public:
    void changeRow(vector<vector<int>>& matrix,int row)//修改行
    {
        for(int c=0;c<matrix[0].size();c++)
        {
            matrix[row][c]=0;
        }
    }
    void changeCol(vector<vector<int>>& matrix,int col)//修改列
    {
        for(int r=0;r<matrix.size();r++)
        {
            matrix[r][col]=0;
        }
    }
    void setZeroes(vector<vector<int>>& matrix) {
        vector<vector<int>> zero;
        int row = matrix.size();
        int col = matrix[0].size();
        for(int r=0;r<row;r++)//遍历数组，寻找零的位置
        {
            for(int c=0;c<col;c++)
            {
                if(matrix[r][c]==0)
                {
                    zero.push_back({r,c});
                }
            }
        }
        for(int i=0;i<zero.size();i++)//修改每行每列
        {
            int zr = zero[i][0];
            int zc = zero[i][1];
            changeRow(matrix,zr);
            changeCol(matrix,zc);
        }
    }
};

运行结果：