【力扣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;
        }
    }
};
相关推荐
明月看潮生2 分钟前
青少年编程与数学 02-016 Python数据结构与算法 28课题、图像处理算法
图像处理·python·算法·青少年编程·编程与数学
_GR13 分钟前
2025年蓝桥杯第十六届C&C++大学B组真题及代码
c语言·数据结构·c++·算法·贪心算法·蓝桥杯·动态规划
心想事“程”13 分钟前
决策树详解+面试常见问题
算法·决策树·机器学习
照海19Gin2 小时前
数据结构中的宝藏秘籍之广义表
c语言·数据结构·算法
小oo呆2 小时前
【自然语言处理与大模型】模型压缩技术之剪枝
算法·机器学习·剪枝
bloxd yzh2 小时前
筛选法(埃氏筛法)C++
数据结构·算法
拓端研究室TRL2 小时前
Python+AI提示词比特币数据预测:Logistic逻辑回归、SVC及XGB特征工程优化实践
开发语言·人工智能·python·算法·逻辑回归
wuqingshun3141594 小时前
蓝桥杯 10.拉马车
数据结构·c++·算法·职场和发展·蓝桥杯·深度优先
我不是小upper5 小时前
详解机器学习各算法的优缺点!!
人工智能·算法·机器学习
不是仙人的闲人5 小时前
算法之动态规划
数据结构·c++·算法·动态规划