Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in place.
题目的意思就是在一个矩阵中找到所有为零的点,然后把该点所在行与列都置为零。但是要求in place,就是原地置换。我理解的意思就是空间复杂度要求o(1)。
很水的一道题,只要遍历整个矩阵记录下为零的点就行了。但是要求空间复杂度o(1)的话可以使用第一行第一列来保留需要的置0的行数与列数。
先判断第一行与第一列是否需要置0,然后遍历矩阵。把需要置0的行数与列数保存在第一行与第一列中。然后把需要置0的行数与列数置0(这里要注意,不要修改第一行与第一列的值),最后通过之前的判断决定是否将第一行第一列置0。
void setZeroes(vector<vector<int> > &matrix)
{
bool firstRow = false;
bool firstColumn = false;
for (auto i : matrix[0])
{
if (i == 0 )
{
firstRow = true;
break;
}
}
for (unsigned int i = 0; i < matrix.size(); i++)
{
if (matrix[i][0] == 0)
{
firstColumn = true;
break;
}
}
for (unsigned int i = 1; i < matrix.size(); i++)
for (unsigned int j = 1; j < matrix[i].size(); j++)
{
if (matrix[i][j] == 0)
{
matrix[0][j] = 0;
matrix[i][0] = 0;
}
}
for (unsigned int i = 0; i < matrix[0].size(); i++)
if (matrix[0][i] == 0)
for (unsigned int j = 1; j < matrix.size(); j++)
matrix[j][i] = 0;
for (unsigned int i = 0; i < matrix.size(); i++)
if (matrix[i][0] == 0)
for (unsigned int j = 1; j < matrix[i].size(); j++)
matrix[i][j] = 0;
if (firstColumn)
for (auto& i : matrix)
i[0] = 0;
if (firstRow)
for (auto& i : matrix[0])
i = 0;
}