【力扣hot100题】(017)矩阵置零

还是挺简单的,使用哈希表记录需要置换的行列即可,这样就可以避免重复节省时间。

cpp 复制代码
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        unordered_set<int> row;
        unordered_set<int> line;
        for(int i=0;i<matrix.size();i++) for(int j=0;j<matrix[0].size();j++) if(matrix[i][j]==0){
            row.insert(i);
            line.insert(j);
        }
        for(auto it=row.begin();it!=row.end();it++){
            for(int i=0;i<matrix[0].size();i++) matrix[*it][i]=0;
        }
        for(auto it=line.begin();it!=line.end();it++){
            for(int i=0;i<matrix.size();i++) matrix[i][*it]=0;
        }
    }
};

看了答案有一种很好的思路,大大缩小空间复杂度,直接用第一行和第一列记录此行/此列要不要归零,因为如果要归零,那么第一行/第一列的这个位置也会归零,非常巧妙。

这个思路还是蛮有难度的,写了半天看答案才写出来。

cpp 复制代码
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool col0=0;
        for(int i=0;i<matrix.size();i++){
            if(matrix[i][0]==0) col0=1;
            for(int j=1;j<matrix[0].size();j++){
                if(matrix[i][j]==0) matrix[0][j]=matrix[i][0]=0;
            }
        }
        for(int i=matrix.size()-1;i>=0;i--){
            for(int j=matrix[0].size()-1;j>0;j--){
                if(matrix[i][0]==0||matrix[0][j]==0) matrix[i][j]=0;
            }
            if(col0==1) matrix[i][0]=0;
        }
    }
};
相关推荐
啊我不会诶27 分钟前
2024CCPC长春邀请赛
算法
珂朵莉MM27 分钟前
第七届全球校园人工智能算法精英大赛-算法巅峰赛产业命题赛第3赛季优化题--启发式算法+操作因子设计
人工智能·算法
CS创新实验室2 小时前
CS实验室行业报告:AI算法工程师就业分析报告
人工智能·算法
XiYang-DING2 小时前
【LeetCode】Hash | 136.只出现一次的数字
算法·leetcode·哈希算法
wayz112 小时前
Day 3:逻辑回归与分类预测
算法·分类·逻辑回归
tankeven2 小时前
HJ176 【模板】滑动窗口
c++·算法
网域小星球3 小时前
C 语言从 0 入门(十二)|指针与数组:数组名本质、指针遍历数组
c语言·算法·指针·数组·指针遍历数组
冰糖拌面3 小时前
二叉树遍历-递归、迭代、Morris
算法
碧海银沙音频科技研究院4 小时前
虚拟机ubuntu与windows共享文件夹(Samba共享)解决WSL加载SI工程满卡问题
人工智能·深度学习·算法
CoovallyAIHub4 小时前
ICLR 2026 | VLM自己学会调检测器:VTool-R1用强化学习教视觉模型使用工具推理
算法·架构·github