LeetCode 73.矩阵置零————2024 春招冲刺百题计划

给定一个 m x n 的矩阵,如果一个元素为 0 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法。

输入:matrix = [[1,1,1],[1,0,1],[1,1,1]]

输出:[[1,0,1],[0,0,0],[1,0,1]]

示例 2:

输入:matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]

输出:[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
思路

方法一:遍历一遍,使用俩set分别存储存在空值的行和列,空间复杂度:O(m+n)

方法二:使用两个标记变量分别记录第一行和第一列是否原本包含 0,用矩阵的第一行和第一列代替方法一中的两个标记数组,以达到 O(1)

方法三:使用一个标记变量记录第一列,而第一列的第一个元素标记第一行,为了防止每一列的第一个元素被提前更新,我们需要从最后一行开始,倒序地处理矩阵元素。
方法一代码

复制代码
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        set<int> s,s1;
        int m = matrix.size();
        int n = matrix[0].size();
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(!matrix[i][j]){
                    s.insert(i);
                    s1.insert(j);
                }
            }
        }
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(s.find(i) != s.end()||s1.find(j) != s1.end())
                    matrix[i][j]=0;
            }
        }

    }
};

方法二代码

复制代码
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();
        bool flag1 = false, flag2 = false;
        for(int i = 0; i < m; i++){
            if(!matrix[i][0]) flag1 = true;
        }
        for(int i = 0; i < n; i++){
            if(!matrix[0][i]) flag2 = true;
        }
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(!matrix[i][j]){
                    matrix[i][0] = matrix[0][j] = 0;
                } 
            }
        }
        for(int i = 1; i < m; i++){
            for(int j = 1; j < n; j++){
                if(!matrix[i][0]||!matrix[0][j]){
                    matrix[i][j] = 0;
                }
            }
        }
        if(flag1){
            for(int i = 0; i < m; i++) matrix[i][0] = 0;
        }
        if(flag2){
            for(int i = 0; i < n; i++) matrix[0][i] = 0;
        }
    }
};
相关推荐
纪元A梦16 小时前
贪心算法应用:K-Means++初始化详解
算法·贪心算法·kmeans
_不会dp不改名_16 小时前
leetcode_21 合并两个有序链表
算法·leetcode·链表
mark-puls16 小时前
C语言打印爱心
c语言·开发语言·算法
Python技术极客16 小时前
将 Python 应用打包成 exe 软件,仅需一行代码搞定!
算法
吃着火锅x唱着歌16 小时前
LeetCode 3302.字典序最小的合法序列
leetcode
睡不醒的kun16 小时前
leetcode算法刷题的第三十四天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
吃着火锅x唱着歌16 小时前
LeetCode 978.最长湍流子数组
数据结构·算法·leetcode
我星期八休息17 小时前
深入理解跳表(Skip List):原理、实现与应用
开发语言·数据结构·人工智能·python·算法·list
lingran__17 小时前
速通ACM省铜第四天 赋源码(G-C-D, Unlucky!)
c++·算法
haogexiaole17 小时前
贪心算法python
算法·贪心算法