若M × N矩阵中某个元素为0,则将其所在的行与列清零
示例:
输入: [ [1,1,1], [1,0,1], [1,1,1] ] 输出: [ [1,0,1], [0,0,0], [1,0,1] ] |
解决思路:
1、循环遍历矩阵,找到值为 0 的元素,并记录其所有值为 0 的位置
2、依次取出值为 0 的位置,将其 行 与 列 置为 0
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int x = matrix.size();
int y = matrix[0].size();
vector<vector<int>> v;
for(int i = 0; i < x; i++)
{
for(int j = 0; j < y; j++)
{
if(matrix[i][j] == 0)
{
v.push_back({i, j});
}
}
}
for(int i = 0; i < v.size(); i++)
{
int vx = v[i][0];
int vy = v[i][1];
for(int vi = 0; vi < y; vi++)
matrix[vx][vi] = 0;
for(int vj = 0; vj < x; vj++)
matrix[vj][vy] = 0;
}
}
};