矩阵置零
程序员文章站
2022-07-15 12:23:52
...
题目
来源:力扣(LeetCode)
链接:矩阵置零
代码
空间复杂度为o(m+n)
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
set<int> rows;//保存含0的行下标
set<int> columns;//保存含0的列下标
for(int i = 0; i < matrix.size(); i++){
for(int j = 0; j < matrix[i].size(); j++){
if(matrix[i][j] == 0){
rows.insert(i);
columns.insert(j);
}
}
}
for(auto index : columns){
for(int i = 0; i < matrix.size(); i++){
matrix[i][index] = 0;
}
}
for(auto index : rows){
for(int j = 0; j < matrix[index].size(); j++){
matrix[index][j] = 0;
}
}
}
};
上一篇: 有效的数独
推荐阅读