1.题目描述
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
示例 1:
示例 2:
2.思路
1.每一行和每一列的第一个元素设置为标识位,如果为0,则标识该行或者该列存在0元素
2.因为第一行和第一列的元素都是matrix[0][0],令其表示第一行,所以单独设置一个flag来标识第一列的情况
3.更新数值时,从右下角到左上角更新,因为标识位位于最左边和最上边,不能影响标识位的值
3.代码
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
bool flag = false;
int m = matrix.size();
int n = matrix[0].size();
for(int i = 0;i < m;++i){
if(matrix[i][0] == 0){//单独判断第一列
flag = true;
}
for(int j = 1;j < n;++j){
if(matrix[i][j] == 0){
matrix[0][j] = matrix[i][0] = 0;
}
}
}
for(int i = m - 1;i >= 0;--i){
for(int j = n - 1;j >= 1;--j){
if(matrix[i][0] == 0 || matrix[0][j] == 0){
matrix[i][j] = 0;
}
}
if(flag){
matrix[i][0] = 0;
}
}
}
};
4.复杂度分析
时间复杂度:O(mn)
空间复杂度:O(1)