1. 题目
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
2. 解题
2.1 O(m+n)空间复杂度
- 用数组记录每行,每列是否有0
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size(), n = matrix[0].size(), i, j;
int r[m] = {0}, c[n] = {0};
for(i = 0; i < m; ++i)
{
for(j = 0; j < n; ++j)
{
if(matrix[i][j] == 0)
{
r[i] = 1;
c[j] = 1;
}
}
}
for(i = 0; i < m; ++i)
{
for(j = 0; j < n; ++j)
{
if(r[i] == 1 || c[j] == 1)
matrix[i][j] = 0;
}
}
}
};
2.2 O(1)空间复杂度
- 利用两个bool变量记录第一行,第一列是否存在0
- 遍历除第一行,第一列外的元素,记录状态在矩阵的第一行,第一列
- 按照第一行,第一列的信息,将右下角矩阵处理掉
- 最后按照bool变量处理第一行,第一列
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
bool r0 = 0, c0 = 0;//记录第1行和第一列是否有0
//遍历除了1行1列外的元素,用第1行1列作为标记位
int i, j, m = matrix.size(), n = matrix[0].size();
for(i = 0; i < m; ++i)
if(matrix[i][0] == 0)
{
c0 = 1;
break;
}
for(j = 0; j < n; ++j)
if(matrix[0][j] == 0)
{
r0 = 1;
break;
}
for(i = 1; i < m; ++i)
{
for(j = 1; j < n; ++j)
{
if(matrix[i][j] == 0)
{
matrix[i][0] = 0;
matrix[0][j] = 0;
}
}
}
for(i = 1; i < m; ++i)
if(matrix[i][0] == 0)
{
for(j = 1; j < n; ++j)
matrix[i][j] = 0;
}
for(j = 1; j < n; ++j)
if(matrix[0][j] == 0)
{
for(i = 1; i < m; ++i)
matrix[i][j] = 0;
}
if(c0)
for(i = 0; i < m; ++i)
matrix[i][0] = 0;
if(r0)
for(j = 0; j < n; ++j)
matrix[0][j] = 0;
}
};