力扣73:矩阵置零

力扣73:矩阵置零

题目

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

思路

题目要求使用原地算法所以我们没法创建一个新矩阵来辅助执行,但是不能创建矩阵不代表我们不能创建数组,我们注意到题目是要让值为0的行和列都变为0那么我们是否可以对要置为0的行列进行标记呢?我们创建两个数组作为标记行和标记列,然后对矩阵进行遍历,遇到值为0时就让它对应的标记行和标记列变为1。然后在遍历结束后再进行一次遍历,这次遍历我们先判断当前行或者列有没有被标记也就是是不是为1如果是就直接让矩阵的值变为0即可。这种方法简单粗暴就是时间复杂度是O(N^2)。

代码

cpp 复制代码
class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int n = matrix.size();    // 多少列
        int m = matrix[0].size(); // 多少行

        vector<int> rows(m, 0); // 行
        vector<int> cols(n, 0); // 列

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (matrix[i][j] == 0) // 第i列第j行为0
                {
                    // 标记这个位置的行和列
                    cols[i] = 1;
                    rows[j] = 1;
                }
            }
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (cols[i] == 1 || rows[j] == 1) {
                    matrix[i][j] = 0;
                }
            }
        }
    }
};
相关推荐
磊-39 分钟前
数组双指针总结
数据结构·算法
.小小陈.1 小时前
链表算法题
数据结构·算法·链表
大飞pkz1 小时前
【算法】排序算法汇总1
开发语言·数据结构·算法·c#·排序算法
CClaris1 小时前
机器学习中的数学——矩阵与向量基础
人工智能·机器学习·矩阵
无风听海1 小时前
神经网络之共现矩阵
人工智能·神经网络·矩阵
前端小L1 小时前
单调栈的“降维打击”:从直方图到矩阵——再探「最大矩形」
数据结构·算法
逐步前行1 小时前
C数据结构--数组|矩阵|广义表
c语言·数据结构·矩阵
小丁努力不焦虑2 小时前
c++基本语法
java·c++·算法
货拉拉技术2 小时前
大模型音频水印技术:用AI守护音频数据的“身份指纹”
人工智能·算法·安全
ysa0510302 小时前
利用数的变形简化大规模问题#数论
c++·笔记·算法