Leetcode—73.矩阵置零【中等】-编程知识

Leetcode—73.矩阵置零【中等】

2023每日刷题（六十六）

Leetcode—73.矩阵置零

在这里插入图片描述

空间复杂度为O(m+n)版实现代码

class Solution {
public:void setZeroes(vector<vector<int>>& matrix) {int rowLen = matrix.size();int colLen = matrix[0].size();vector<int> row(rowLen, 0);vector<int> col(colLen, 0);for(int i = 0; i < rowLen; i++) {for(int j = 0; j < colLen; j++) {if(matrix[i][j] == 0) {row[i] = 1;col[j] = 1;}}}for(int i = 0; i < rowLen; i++) {for(int j = 0; j < colLen; j++) {if(row[i] || col[j]) {matrix[i][j] = 0;}}}}
};

运行结果

在这里插入图片描述

优化空间复杂度为O(1)版算法思想

在这里插入图片描述

优化版实现代码

class Solution {
public:void setZeroes(vector<vector<int>>& matrix) {bool rowFlag = false, colFlag = false;int rowLen = matrix.size();int colLen = matrix[0].size();for(int i = 0; i < rowLen; i++) {if(!matrix[i][0]) {colFlag = true;}}for(int j = 0; j < colLen; j++) {if(!matrix[0][j]) {rowFlag = true;}}for(int i = 1; i < rowLen; i++) {for(int j = 1; j < colLen; j++) {if(!matrix[i][j]) {matrix[i][0] = matrix[0][j] = 0;}}}for(int i = 1; i < rowLen; i++) {for(int j = 1; j < colLen; j++) {if(!matrix[i][0] || !matrix[0][j]) {matrix[i][j] = 0;}}}if(colFlag) {for(int i = 0; i < rowLen; i++) {matrix[i][0] = 0;}}if(rowFlag) {for(int j = 0; j < colLen; j++) {matrix[0][j] = 0;}}}
};