矩阵置零

• 有关题目
• • 题解

有关题目

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

提示：

m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-2^31 <= matrix[i][j] <= 2^31 - 1

题解

法一：使用标记数组
参考官方题解

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int m = matrix.size();
        int n = matrix[0].size();

        vector<int> row(m), col(n);//标记数组，记录每一行，每一列是否有为零的元素
        
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (0 == matrix[i][j])//当前行和列出现为0的元素，将该行和该列标记
                {
                    row[i] = col[j] = 1;
                }
            }
        }

        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (row[i] || col[j])//该行或该列被标记，对应的元素置零
                    matrix[i][j] = 0;
            }
        }
    }
};