LeetCode - 零矩阵

题目来源：https://leetcode-cn.com/problems/zero-matrix-lcci/

题目描述：

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

]

思路：

直接遍历数组，遇到0时将给元素所在的行和列置0。这样会导致多余操作，误将置0后的行列也再次置0，最后可能导致整个数组都为0。

可以使用标记存储0的位置，遍历完一遍后再进行置0操作。

这里给出的是使用标记数组记录位置的方法，比较简单易懂，但需要调用较多的额外空间O(mn)。

官方题解还给出了更好的方法，使用两个flag标记变量来处理，或者直接使用一个标记变量来处理。

附上链接，供参考：https://leetcode-cn.com/problems/zero-matrix-lcci/solution/ling-ju-zhen-by-leetcode-solution-7ogg/

代码：

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        vector<int> row(m),line(n);	//记录0的行和列
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(matrix[i][j] == 0){
                    row[i] = 1,line[j] = 1;
                }
            }
        }
        for(int i=0;i<m;i++){	//将整行置0
            if(row[i] == 1){
                for(int j=0;j<n;j++) matrix[i][j] = 0;
            }
        }
        for(int j=0;j<n;j++){	//将整列置0
            if(line[j] == 1){
                for(int i=0;i<m;i++) matrix[i][j] = 0;
            }
        }
    }
};