使用一个数组标记矩阵值为0的横坐标和纵坐标,然后将所在行和列置零
class Solution {
public:
void makeZero(pair<int, int>& pos, vector<vector<int>>& matrix){
auto [x, y] = pos;
for(int i = 0; i < matrix.size(); i++){
matrix[i][y] = 0;
}
for(int j = 0; j < matrix[0].size(); j++){
matrix[x][j] = 0;
}
}
void setZeroes(vector<vector<int>>& matrix) {
vector<pair<int, int>> v;
for(int i = 0; i < matrix.size(); i++){
for(int j = 0; j < matrix[0].size(); j++){
if(matrix[i][j] == 0){
v.push_back(make_pair(i, j));
}
}
}
for(auto& pos : v){
makeZero(pos, matrix);
}
}
};
官方题解单独处理第0行和第0列,然后利用第0行和第0列来标记是否置零,优化了空间复杂度
class Solution {
public:
void setZeroes(vector<vector<int>>& matrix) {
int m = matrix.size();
int n = matrix[0].size();
int flag_col0 = false, flag_row0 = false;
for (int i = 0; i < m; i++) {
if (!matrix[i][0]) {
flag_col0 = true;
}
}
for (int j = 0; j < n; j++) {
if (!matrix[0][j]) {
flag_row0 = true;
}
}
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;
}
}
}
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;
}
}
}
if (flag_col0) {
for (int i = 0; i < m; i++) {
matrix[i][0] = 0;
}
}
if (flag_row0) {
for (int j = 0; j < n; j++) {
matrix[0][j] = 0;
}
}
}
};
作者:力扣官方题解
链接:https://leetcode.cn/problems/set-matrix-zeroes/solutions/669901/ju-zhen-zhi-ling-by-leetcode-solution-9ll7/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。