
还是挺简单的,使用哈希表记录需要置换的行列即可,这样就可以避免重复节省时间。
cpp
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
unordered_set<int> row;
unordered_set<int> line;
for(int i=0;i<matrix.size();i++) for(int j=0;j<matrix[0].size();j++) if(matrix[i][j]==0){
row.insert(i);
line.insert(j);
}
for(auto it=row.begin();it!=row.end();it++){
for(int i=0;i<matrix[0].size();i++) matrix[*it][i]=0;
}
for(auto it=line.begin();it!=line.end();it++){
for(int i=0;i<matrix.size();i++) matrix[i][*it]=0;
}
}
};
看了答案有一种很好的思路,大大缩小空间复杂度,直接用第一行和第一列记录此行/此列要不要归零,因为如果要归零,那么第一行/第一列的这个位置也会归零,非常巧妙。
这个思路还是蛮有难度的,写了半天看答案才写出来。
cpp
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
bool col0=0;
for(int i=0;i<matrix.size();i++){
if(matrix[i][0]==0) col0=1;
for(int j=1;j<matrix[0].size();j++){
if(matrix[i][j]==0) matrix[0][j]=matrix[i][0]=0;
}
}
for(int i=matrix.size()-1;i>=0;i--){
for(int j=matrix[0].size()-1;j>0;j--){
if(matrix[i][0]==0||matrix[0][j]==0) matrix[i][j]=0;
}
if(col0==1) matrix[i][0]=0;
}
}
};