【力扣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;
        }
    }
};
相关推荐
Kaltistss29 分钟前
98.验证二叉搜索树
算法·leetcode·职场和发展
知己如祭33 分钟前
图论基础(DFS、BFS、拓扑排序)
算法
mit6.82441 分钟前
[Cyclone] 哈希算法 | SIMD优化哈希计算 | 大数运算 (Int类)
算法·哈希算法
c++bug44 分钟前
动态规划VS记忆化搜索(2)
算法·动态规划
哪 吒1 小时前
2025B卷 - 华为OD机试七日集训第5期 - 按算法分类,由易到难,循序渐进,玩转OD(Python/JS/C/C++)
python·算法·华为od·华为od机试·2025b卷
军训猫猫头1 小时前
1.如何对多个控件进行高效的绑定 C#例子 WPF例子
开发语言·算法·c#·.net
success2 小时前
【爆刷力扣-数组】二分查找 及 衍生题型
算法
Orlando cron2 小时前
数据结构入门:链表
数据结构·算法·链表
牛客企业服务3 小时前
2025年AI面试推荐榜单,数字化招聘转型优选
人工智能·python·算法·面试·职场和发展·金融·求职招聘
糖葫芦君4 小时前
Policy Gradient【强化学习的数学原理】
算法