Leetcode_c++：Set Matrix Zeros (073）

题目

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.

---

算法

复杂度：O(n+m)

先通过两个数组来记录该行该列上有没有零元素，等对矩阵遍历完之后再统一修改即可。

1）创建一个两个数组，分别用于记录矩阵中的行和列是否需要变为0的标志；
2）遍历矩阵，标记矩阵中哪些行和哪些列元素需要变为0；
3）遍历矩阵，根据矩阵中对应行列是否为0标志来更改矩阵中的元素即可。

---

class Solution
{
public:
    void setZeros(vector< vector<int> > &matrix)
    {
        // 获取矩阵的行数 
        int row = matrix.size();
        if (row < 1)
            return ;
        // 获取矩阵的列数 
        int col = matrix[0].size();
        // 定义存放行列是否变0标志数组 
        vector<bool> colRecorder(col, false);
        vector<bool> rowRecorder(row, false);
        // 遍历矩阵，标志哪些行和哪些列需要变为0 
        for (int i = 0; i < row; i++)
        {
            for (int j = 0; j < col; j++)
            {
                if (matrix[i][j] == 0)
                {
                    rowRecorder[i] = true;
                    colRecorder[j] = true;
                }
            }
        }
        // 遍历矩阵，根据行列是否为0标志对矩阵中的数据进行更改 
        for (int i = 0; i < row; i++)
        {
            for (int j = 0; j < col; j++)
            {
                if (rowRecorder[i])
                    matrix[i][j] = 0;
                else if (colRecorder[j])
                    matrix[i][j] = 0;
            }
        }
    }
};