Leetcode Set Matrix Zeroes

看了网上的题解，很巧妙的做法。

本来想到的做法是O(m+n)，用两个数组来分别记录哪一行和哪一列需要置0。

而O（1）的做法实际上就是利用矩阵的第一行和第一列来充当这两个数组，当然事先要判断一下，第一行和第一列要不要置0，作上标记。到最后再来更新。

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        bool rtag = false, ctag = false;
        int row = matrix.size(), col = matrix[0].size();
        for(int i = 0; i < col; i++){
            if(matrix[0][i] == 0){
                rtag = true;
                break;
            }
        }
        for(int i = 0; i < row; i++){
            if(matrix[i][0] == 0){
                ctag = true;
                break;
            }
        }
        for(int i = 1; i < row; i++){
            for(int j = 1; j < col; j++){
                if(matrix[i][j] == 0){
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }
        
        for(int i = 1; i < col; i++){
            if(matrix[0][i] == 0){
                for(int j = 1; j < row; j++){
                    matrix[j][i] = 0;
                }
            }
        }
        for(int i = 1; i < row; i++){
            if(matrix[i][0] == 0){
                for(int j = 1; j < col; j++)
                    matrix[i][j] = 0;
            }
        }
        if(rtag){
            for(int i = 0; i < col; i++)
                matrix[0][i] = 0;
        }
        if(ctag){
            for(int i = 0; i < row; i++)
                matrix[i][0] = 0;
        }
    }
};