力扣73:矩阵置零

力扣73:矩阵置零

• 题目
• 思路
• 代码

题目

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

思路

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

代码

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;
                }
            }
        }
    }
};