每日一算-73. 矩阵置零

题目描述：
给定一个 m x n 的矩阵，如果一个元素为 0，则将其所在行和列的所有元素都设为 0。请使用原地算法。

示例 1:

输入:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
输出:
[
[1,0,1],
[0,0,0],
[1,0,1]
]

解题思路：
题目中要求使用原地算法，
方法一，可以用空间O(n + m)的算法，使用一个辅助矩阵，记录要置0的行和列
方法二，使用空间O(2)的算法，使用两个辅助空间记录首行和首列是否要置0，然后使用首行和首列的空间记录每一行、每一列是否要置0
cpp代码：
class Solution {
public:
void setZeroes(vector<vector>& matrix) {
if(matrix.empty()) return ;
if(matrix[0].empty()) return;
bool row = false;
bool col = false;
for(int i = 0;i < matrix.size();i++) {
if(matrix[i][0] == 0) {
col = true;
break;
}
}
for(int j = 0;j < matrix[0].size();j++) {
if(matrix[0][j] == 0) {
row = true;
break;
}
}
for(int i = 1;i < matrix.size();i++){
for(int j = 1;j < matrix[0].size();j++){
if(matrix[i][j] == 0) {
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(int i = 1;i < matrix.size();i++) {
if(matrix[i][0] == 0){
for(int j = 1;j < matrix[i].size();j++) matrix[i][j] = 0;
}
}
for(int i = 1;i < matrix[0].size();i++) {
if(matrix[0][i] == 0) {
for(int j = 1;j < matrix.size();j++) matrix[j][i] = 0;
}
}
if(row == true) {
for(int i = 0;i < matrix[0].size();i++) matrix[0][i] = 0;
}
if(col == true){
for(int i = 0;i < matrix.size();i++) matrix[i][0] = 0;
}
return;
}
};