Leetcode 73.矩阵置零

题目：

给定一个 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]]

提示：

• m == matrix.length
• n == matrix[0].length
• 1 <= m, n <= 200
• -231 <= matrix[i][j] <= 231 - 1

题解：

解法一：

实际上就是多开两个数组row和col，分别记录哪些行和列需要置零，遍历矩阵中的每个元素，如果某个元素为0，将对应的row和col数组中的元素置为1，表示对应的行和列需要置零。多开了两个 数组，空间复杂度是 O(m+n)了。 

代码如下：

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
    int m=matrix.size();
    int n=matrix[0].size();
    vector<int> row(m,0),col(n,0);
    for(int i=0;i<m;i++) {
        for(int j=0;j<n;j++) {
            if(matrix[i][j]==0) {
                row[i]=1;
                col[j]=1;
            }
        }
    }
    for(int i=0;i<m;i++) {
        for(int j=0;j<n;j++) {
            if(row[i]==1||col[j]==1) {
                matrix[i][j]=0;
            }
        }
    }
    }
};

解法二：

实际上和上面的做法是一样的，不同的是没有再用额外的空间，而是用矩阵的第一行和第一列来表示相应行和列是否需要置零。

遍历矩阵中的每个元素，如果某个元素为0，将需要置零的行和列的信息保存到第一行和第一列中，根据第一行和第一列的信息，我们更新整个矩阵，该置零的置零。

代码如下：

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        bool firstRowHasZero = false;
        bool firstColHasZero = false;

        for (int j = 0; j < n; j++) {
            if (matrix[0][j] == 0) {
                firstRowHasZero = true;
                break;
            }
        }

        for (int i = 0; i < m; i++) {
            if (matrix[i][0] == 0) {
                firstColHasZero = true;
                break;
            }
        }

        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][j] == 0) {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }

        for (int i = 1; i < m; i++) {
            for (int j = 1; j < n; j++) {
                if (matrix[i][0] == 0 || matrix[0][j] == 0) {
                    matrix[i][j] = 0;
                }
            }
        }

        if (firstRowHasZero) {
            for (int j = 0; j < n; j++) {
                matrix[0][j] = 0;
            }
        }

        if (firstColHasZero) {
            for (int i = 0; i < m; i++) {
                matrix[i][0] = 0;
            }
        }
    }
};