给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用
#include<iostream>
#include<vector>
#include<cmath>
#include<algorithm>
#include<string>
using namespace std;
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
vector<int> row;
vector<int> col;
for(int i = 0; i < matrix.size(); i++){
for(int j = 0; j < matrix[i].size(); j++){
if (matrix[i][j] == 0){
row.push_back(i);
col.push_back(j);
}
}
}
for(int i = 0; i < row.size(); i++){
for(int j = 0; j < matrix[0].size();j++){
matrix[row[i]][j] = 0;
}
}
for(int i = 0; i < col.size(); i++){
for(int j = 0; j < matrix.size();j++){
matrix[j][col[i]] = 0;
}
}
}
};
int main() {
return 0;
}