Problem
Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.
Example 1:
Input:
[
[1,1,1],
[1,0,1],
[1,1,1]
]
Output:
[
[1,0,1],
[0,0,0],
[1,0,1]
]
Example 2:
Input:
[
[0,1,2,0],
[3,4,5,2],
[1,3,1,5]
]
Output:
[
[0,0,0,0],
[0,4,5,0],
[0,3,1,0]
]
Follow up:
- A straight forward solution using O(mn) space is probably a bad idea.
- A simple improvement uses O(m + n) space, but still not the best solution.
- Could you devise a constant space solution?
解题思路
思路一:
这一题有一个很自然的想法,首先我们遍历矩阵中的所有数据,如果当前数据是0,则分别将它的行号和列号保存在一个数组中。第二次遍历所有的数据时,我们在数组中分别查找行号和列号,如果找到了对应的行号或者列号则将数据变成0。代码如下:(需要注意的是这种方法需要额外的空间保存行号和列号,因此空间复杂度是O(m + n)。m表示行数,n表示列数)。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
set<int> rows;
set<int> cols;
for (size_t i = 0; i < matrix.size(); ++i)
{
for (size_t j = 0; j < matrix[0].size(); ++j)
{
if (matrix[i][j] == 0)
{
rows.insert(i);
cols.insert(j);
}
}
}
for (size_t i = 0; i < matrix.size(); ++i)
{
for (size_t j = 0; j < matrix[0].size(); ++j)
{
if (rows.find(i) != rows.end() || cols.find(j) != cols.end())
{
matrix[i][j] = 0;
}
}
}
}
};
思路二:
在问题中有提出能不能使用常数级别的空间复杂度解决问题,但是我们依然还是要解决的一个问题是行号和列号的保存。如此一来,我们考虑到能不能用每一行和每一列的第一个数据来保存这一行存不存在数据0,然后遍历的时候查找当前数据的行的第一个数据和这一列的第一个数据即可。这里需要说明的一点是,如果用来保存数据的第一行和第一列存在0的话,情况会变成不管怎么样第一行或者第一列都会变成0,因此我们需要两个变量来保存第一行和第一列是不是存在数据0。而这也是唯一需要的额外的空间,代码如下。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
bool row = true;
bool col = true;
for (size_t i = 0; i < matrix.size(); ++i)
{
if (matrix[i][0] == 0)
{
row = false;
}
}
for (size_t i = 0; i < matrix[0].size(); ++i)
{
if (matrix[0][i] == 0)
{
col = false;
}
}
for (size_t i = 0; i < matrix.size(); ++i)
{
for (size_t j = 0; j < matrix[0].size(); ++j)
{
if (matrix[i][j] == 0)
{
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for (size_t i = 1; i < matrix.size(); ++i)
{
for (size_t j = 1; j < matrix[0].size(); ++j)
{
if (matrix[i][0] == 0 || matrix[0][j] == 0)
{
matrix[i][j] = 0;
}
}
}
if (!row)
{
for (size_t i = 0; i < matrix.size(); ++i)
{
matrix[i][0] = 0;
}
}
if (!col)
{
for (size_t i = 0; i < matrix[0].size(); ++i)
{
matrix[0][i] = 0;
}
}
}
};