题目描述:
给定一个 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;
}
};