1、题目
给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。
链接:https://leetcode-cn.com/problems/set-matrix-zeroes/
2、思路分析
1)双标记法
这个方法的巧妙之处在于将原本需要用两个数组保存的数字0的位置,它将其放到了数组的第一行和第一列。
先用双标记,记录如果第一行和第一列存在0,则标记为真。
再遍历数组,如果存在0,那么将这行和列的第一个元素置为0。
下一步就是继续遍历数组,如果该行和列的第一个元素为0,那么将该位置置为0 。
最后就是如果标记为真,那么将该行和列置为0 。
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
// 双标记法
int m = matrix.size();
int n = matrix[0].size();
int fr = false, fc = false;
// 确定第一行和第一列是否有0
for(int i = 0; i < m; ++i){
if(!matrix[i][0]){
fr = true;
}
}
for(int i = 0; i < n; ++i){
if(!matrix[0][i]){
fc = true;
}
}
// 将除了第一行和第一列中有0的元素的该行和该列的第一个元素置为0
for(int i = 1; i < m; ++i){
for(int j = 1; j < n; ++j){
if(!matrix[i][j]){
matrix[i][0] = matrix[0][j] = 0;
}
}
}
// 如果该行或列第一个元素为0,则将该位置置为0
for(int i = 1; i < m; ++i){
for(int j = 1; j < n; ++j){
if(!matrix[i][0] || !matrix[0][j]){
matrix[i][j] = 0;
}
}
}
// 将标记的行和列置为0
for(int i = 0; i < m; ++i){
if(fr){
matrix[i][0] = 0;
}
}
for(int i = 0; i < n; ++i){
if(fc){
matrix[0][i] = 0;
}
}
}
};