【力扣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;
        }
    }
};
相关推荐
不是吧这都有重名43 分钟前
[论文阅读]Deeply-Supervised Nets
论文阅读·人工智能·算法·大语言模型
homelook1 小时前
matlab simulink双边反激式变压器锂离子电池均衡系统,双目标均衡策略,仿真模型,提高均衡速度38%
算法
什码情况2 小时前
星际篮球争霸赛/MVP争夺战 - 华为OD机试真题(A卷、Java题解)
java·数据结构·算法·华为od·面试·机试
天上路人2 小时前
采用AI神经网络降噪算法的通信语音降噪(ENC)模组性能测试和应用
人工智能·神经网络·算法
字节高级特工2 小时前
【C++】”如虎添翼“:模板初阶
java·c语言·前端·javascript·c++·学习·算法
.Vcoistnt2 小时前
Codeforces Round 1024 (Div. 2)(A-D)
数据结构·c++·算法·贪心算法·动态规划·图论
晴天下小雨o2 小时前
排序算法总结
java·算法·排序算法
程序员爱钓鱼2 小时前
循环语句:for、range -《Go语言实战指南》
java·数据结构·算法
LabVIEW开发3 小时前
LabVIEW中算法开发的系统化解决方案与优化
算法·labview
chenyuhao20243 小时前
链表面试题7之相交链表
数据结构·算法·链表·面试·c#