73. 矩阵置零

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

73. 矩阵置零_第1张图片

本题的难点在与如果找到了矩阵中的0,并修改与他的行列为0;呢么下次遍历的时候就弄乱了之前的0的位置,基于数字孪生的思想,采用一个复制矩阵,在复制矩阵上找0的位置,并同时修该原矩阵。

class Solution {
public:
    void setZeroes(vector>& matrix) {
        if (matrix.size() == 0 || matrix[0].size() == 0) return;
        vector> newM(matrix);
        const int M = matrix.size(), N = matrix[0].size();
        for (int r = 0; r < M; ++r) {
            for (int c = 0; c < N; ++c) {
                if (newM[r][c] == 0) {
                    for (int j = 0; j < N; ++j) {
                        matrix[r][j] = 0;
                    }
                    for (int i = 0; i < M; ++i) {
                        matrix[i][c] = 0;
                    }
                }
            }
        }
    }
};

你可能感兴趣的:(矩阵,人工智能,python)