难度级别(中等)
-给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
进阶:
一个直观的解决方案是使用 O(mn) 的额外空间,但这并不是一个好的解决方案。 一个简单的改进方案是使用 O(m + n) 的额外空间,但这仍然不是最好的解决方案。 你能想出一个仅使用常量空间的解决方案吗?
- 题解代码如下
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
///如果行或者列为空则不需要做任何操作直接返回
if(matrix.empty() || matrix[0].empty()) return;
///用n表示行,用m表示列
int n = matrix.size(), m = matrix[0].size();
///用r0,c0表示第0行和第0列的信息
int r0 = 1, c0 = 1;
///如果0行有为0的话记为0
for(int i = 0;i < m;i ++){
if(matrix[0][i] == 0){
r0 = 0;
}
}
///如果第0列中有0的话记为0
for(int i = 0;i < n;i ++){
if(matrix[i][0] == 0){
c0 = 0;
}
}
///看一下每一行是不是需要生成0
for(int i = 1;i < n;i ++){
for(int j = 0;j < m;j++){
if(matrix[i][j] == 0){
matrix[i][0] = 0;
}
}
}
///看一下每一列是不是需要生成0
for(int i=1;i<m;i++){
for(int j=0;j<n;j++){
if(matrix[j][i]==0){
matrix[0][i] = 0;
}
}
}
///把所有标记的行全部刷成0
for(int i=1;i<n;i++){
if(matrix[i][0] == 0){
for(int j=0;j<m;j++){
matrix[i][j]=0;
}
}
}
///把所有标记的列全部刷成0
for(int i=1;i<m;i++){
if(matrix[0][i]==0){
for(int j=0;j<n;j++){
matrix[j][i]=0;
}
}
}
///处理第0行第0列
if(!r0)
for(int i=0;i<m;i++){
matrix[0][i]=0;
}
if(!c0)
for(int i=0;i<n;i++){
matrix[i][0]=0;
}
}
};