【力扣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;
        }
    }
};
相关推荐
闲研随记9 分钟前
【文献阅读 ICLR 2026】RL算法:Co-Rewarding
算法
青山木18 分钟前
Hot 100 --- 最小栈
java·数据结构·算法·leetcode
疯狂打码的少年21 分钟前
【数据结构】栈的应用:表达式求值(后缀表达式)
java·数据结构·笔记·算法
HiDev_32 分钟前
【非标自动化】plc执行顺序问题、扫描周期问题
深度学习·算法·机器学习
高洁011 小时前
工信部教考中心证书-人工智能系列本系列
人工智能·python·深度学习·算法
陈王卜1 小时前
Reduce 算子优化笔记
算法
wabs6661 小时前
关于图论【最短路径之Bellman_ford 算法(判断负权回路)|卡码网95.城市间货物运输II的思考】
数据结构·算法·图论·bellman_ford算法·卡码网·判断负权回路
YouonlyliveonceC2 小时前
《大话数据结构》第4章实战:中缀表达式转后缀 + 后缀表达式求值(完整可运行实现)
算法
phoenix@Capricornus2 小时前
从统计决策到贝叶斯估计
人工智能·算法·机器学习
AICDragon2 小时前
1.8%就够了:K3的896个MoE专家,为什么激活率这么低反而是好事?
人工智能·算法