Set Matrix Zeroes

1，题目要求

Given a m x n matrix, if an element is 0, set its entire row and column to 0. Do it in-place.

Example 1:
Input:

[
  [1,1,1],
  [1,0,1],
  [1,1,1]
]

Output:

[
  [1,0,1],
  [0,0,0],
  [1,0,1]
]

Example 2:
Input:

[
  [0,1,2,0],
  [3,4,5,2],
  [1,3,1,5]
]

Output:

[
  [0,0,0,0],
  [0,4,5,0],
  [0,3,1,0]
]

Follow up:

• A straight forward solution using O(mn) space is probably a bad idea.
• A simple improvement uses O(m + n) space, but still not the best solution.
• Could you devise a constant space solution?

给定m×n矩阵，如果元素为0，则将其整个行和列设置为0.就地执行。

2，题目思路

对于这道题，要求将一个矩阵中含有0的位置设定该行和该列都设为0。

这道题有点类似于炸弹人的游戏机制，炸弹将一行和一列全部炸裂了。
说到这道题，如果没有空间的限制，很容易实现。
如果是O(mn)，非常简单，新开辟一个新的矩阵即可；
如果是O(m+n)，也不是很困难，我们利用额外的空间记录对应的行和列是否为0即可。
然而，如果是常量空间复杂度，就比较麻烦了。
就像之前做过的一题Game of Life，这时，我们需要在矩阵上记录结果。

因此，在这道题的解决上，我们的思路是在矩阵的第一行和第一列来记录该行是否为0。
但是有一点，如果我们从左上角（0，0）开始遍历，会有一个问题：因为row0的状态和column0的状态将占用相同的单元格——matrix[0][0]，即加入说matrix[2][0]为0，会使得matrix[0][0]和matrix[2][0]同时设置为0，但是如果第一列没有为0的数字，在判断上就会出现错误。

因此，我们需要对第一列单独判断，即：

• i : 0 ~ n
• j：1~m

第一步：自上而下、自左向右进行判断；
第二步：自下而上、自右向左根据第一行和第一列的状态，对matrix[i][j]的值进行设置。

最后，最后单独对第一列进行判断即可。

3，代码实现

auto x = []() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    return 0;
}();

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        bool isCol0Zero = false;
        int rows = matrix.size(), cols = matrix[0].size();
        
        for(int i = 0;i<rows;i++){
            if(matrix[i][0] == 0)
                isCol0Zero = true;   //说明第一列为0
            for(int j = 1;j<cols;j++)
                if(matrix[i][j] == 0)
                    matrix[i][0] = matrix[0][j] = 0;
        }
        
        for(int i = rows-1;i>=0;i--){
            for(int j = cols-1;j>=1;j--)
                if(matrix[i][0] == 0 || matrix[0][j] == 0)
                    matrix[i][j] = 0;
            if(isCol0Zero)
                matrix[i][0] = 0;
        }
    }
};