当前位置：首页 > news >正文

leetcode hot100刷题日记——9.矩阵置零

news 来源：原创 2025/5/23 9:19:21

在这里插入图片描述

解答一：

class Solution {
public:void setZeroes(vector<vector<int>>& matrix) {//O(mn)额外空间,直白思想：找个和matrix一样形状的数组做vis标记数组//如果遇到了0，那就vis[i][j]=1,表明这个数组位置已经被用过了//不会出现：如示例1中，遍历到第二行第二列0时，把第二行和第二列所有数字变成0//下一个遍历第二行第三列，已经在上一步中变成0了//你要是不做标记，那也会把第三列所有数变成0，那就错了//但是，注意示例2又提醒了我们另一种情况//当两个0在同一行时候，如果你在遍历到第二个0的时候，vis是1//那么他就会跳过这个0，不把第四列变成0了//所以在进行vis更新前，我们需要判断一下这个数是不是原本就是0//如果原本就是0的话，vis就依旧是0int m=matrix.size();int n=matrix[0].size();vector<vector<int>> vis(m,vector<int>(n));for(int i=0;i<m;i++){for(int j=0;j<n;j++){if(matrix[i][j]==0&&vis[i][j]==0){for(int k=0;k<n;k++){if(matrix[i][k]!=0){vis[i][k]=1;}matrix[i][k]=0; }for(int k=0;k<m;k++){if(matrix[k][j]!=0){vis[k][j]=1;}matrix[k][j]=0;}}}}return;}
};

时间复杂度：O(mn)
空间复杂度：O(mn)

解答二：

class Solution {
public:void setZeroes(vector<vector<int>>& matrix) {//O(m+n)额外空间：一个数组存行位置，一个数组存列位置int mm=matrix.size();int nn=matrix[0].size();vector<int> m(mm);vector<int> n(nn);for(int i=0;i<mm;i++){for(int j=0;j<nn;j++){//把所有是0的数的行位置和列位置记录下来if(matrix[i][j]==0){m[i]=n[j]=1;}}}for(int i=0;i<mm;i++){for(int j=0;j<nn;j++){if(m[i]!=0||n[j]!=0){//只要行有0或者列有0就直接变0matrix[i][j]=0;}}}}
};