set具有元素唯一性,insert之后如果有无所谓了,查看leetcode 73题
void setZeroes(vector<vector<int>>& matrix) {
set<int> row;
set<int> col;
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);
col.insert(j);
}
}
}
for (auto i: row) {
for (int j = 0; j < matrix[0].size(); j++) {
matrix[i][j] = 0;
}
}
for (int i = 0; i < matrix.size(); i++) {
for (auto j: col) {
matrix[i][j] = 0;
}
}
}