Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
思路:先把行列中有0的记录下来,再把这些行列归零,两步分开做
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix)
{
if (matrix.size() == 0)
return;
int row = matrix.size();
int col = matrix[0].size();
int r[row] = {0};
int c[col] = {0};
for (int i = 0; i < row; i++)
{
for (int j = 0; j < col; j++)
{
if (matrix[i][j] == 0)
{
r[i] = 1;
c[j] = 1;
}
}
}
for (int i = 0; i < row; i++)
{
if(r[i] == 1)
for(int j = 0; j < col; j++)
matrix[i][j] = 0;
}
for (int j = 0; j < col; j++)
{
if (c[j] == 1)
for (int i = 0; i < row; i++)
matrix[i][j] = 0;
}
return;
}
};