73. Set Matrix Zeroes（矩阵置零）解法（C++ & 注释）

1. 题目描述

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

示例 1:

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

示例 2:

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

进阶:

• 一个直接的解决方案是使用 O(mn) 的额外空间，但这并不是一个好的解决方案。
• 一个简单的改进方案是使用 O(m + n)的额外空间，但这仍然不是最好的解决方案。
• 你能想出一个常数空间的解决方案吗？

题目链接：中文题目；英文题目

2. 标记（Marks）

2.1 解题思路

这道题最核心的思路就是标记哪些行和列需要置零，最简单的方法就是用一个额外数组存储需要置零的行和列，遍历matrix后，把这些行和列置零即可，代码也不难写。

所以这里只讲解不需要使用的额外空间的思路，其他方法，大家可以看看参考资料里面的文章，这里就不再赘述了。

如果不需要额外空间进行标注行和列，那就需要把标记放在matrix里面了。通过观察，我们可以发现下面的规律（下面matrix[i][j] == 0）：

1. i == 0 && j == 0，第一行（matrix[0][j]）和第一列（matrix[i][0]）需要置零；
2. i == 0，当前列（matrix[i][j]）需要置零，第一行（matrix[0][j]）需要置零；
3. j == 0，当前行（matrix[i][j]）需要置零，第一列（matrix[i][0]）需要置零；
4. 其余情况，当前行（matrix[i][j]）和当前列（matrix[i][j]）需要置零；

所以4. 其余情况，我们可以用第一行和第一列来标记，即：

if (!matrix[i][j]) {matrix[0][j] = 0, matrix[i][0] = 0;}

现在第一行和第一列（原点除外）失去了处理自己所属列和所属行的功能，所以我们对1. 2. 3. 分别进行处理，我们用right和down分别标记第一行和第一列是否需要置零：

1. 原点：right 和 down 为1，表示第一行和第一列都需要置零；
2. 第一列：matrix[0][j]已经表示当前列需要置零，只需right = 1，表示第一行置零；
3. 第一行：matrix[i][0]已经表示当前行需要置零，只需down = 1，表示第一列置零；

2.2 示例代码

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

        for (int i = 0; i < matrix.size(); i++) {
            for (int j = 0; j < matrix[0].size(); j++) {
                if (!i && !j && !matrix[i][j]) { right = 1; down = 1; } // 原点
                else if (!j && !matrix[i][j]) down = 1; // 左边界
                else if (!i && !matrix[i][j]) right = 1; // 上边界
                else if (!matrix[i][j]) { matrix[i][0] = 0; matrix[0][j] = 0; }
            }
        }
         
        // 置零操作
        for (int i = 1; i < m; i++)
            if (!matrix[i][0]) for (int j = 1; j < n; j++) matrix[i][j] = 0;

        for (int j = 1; j < n; j++)
            if (!matrix[0][j]) for (int i = 1; i < m; i++) matrix[i][j] = 0;

        if (right) for (int j = 0; j < n - 1; j++) matrix[0][j] = 0;
        if (down) for (int i = 0; i < m - 1; i++) matrix[i][0] = 0;
    }
};

3. 参考资料

1. 73. Set Matrix Zeroes(英文)
2. 73. Set Matrix Zeroes(中文)

---