leetcode 73. Set Matrix Zeroes

题意

题解

代码

class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
set<int> rows, cols;
int m = matrix.size(), n = matrix[0].size();
for(int i = 0; i < m; i++)
{
for(int j = 0; j < n; j++)
{
if(matrix[i][j] == 0)
rows.insert(i), cols.insert(j);
}

}

for(set<int>::iterator i = rows.begin(); i != rows.end(); i++)
{
for(int j = 0; j < n; j++)
{
matrix[*i][j] = 0;
}

}
for(set<int>::iterator i = cols.begin(); i != cols.end(); i++)
{
for(int j = 0; j < m; j++)
{
matrix[j][*i] = 0;
}

}

}
};