LeetCode--Set Matrix Zeroes

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

click to show follow up.

Follow up:
Did you use extra space?
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?

思路：标志位。这个题有三种层次的解法：
第一，复制一个新二维数组，遍历原二维数组，有0的时候新二维数组的那一行和那一列就为0；
第二，开辟一个大小为m的一维数组和大小为n的一维数组，分别记录第m行和第n列是否有数组为0，最后在设置相应行列为0；
第三，用第一行和第一列分别来记录每一列和每一行是否有0，然后再设置相应行列都为0，不过要注意先判断第一行和第一列的情况，用两个变量记录，如果第一行和第一列有0，最后第一行和第一列则全为0.

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        if(matrix.empty()||matrix[0].empty()) return;
        int m=matrix.size(),n=matrix[0].size();
        bool rowZero=false,colZero=false;
        for(int i=0;i<m;i++){
            if(matrix[i][0]==0) colZero=true;
        }
        for(int i=0;i<n;i++){
            if(matrix[0][i]==0) rowZero=true;
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(matrix[i][j]==0){
                    matrix[0][j]=0;
                    matrix[i][0]=0;
                }
            }
        }
        for(int i=1;i<m;i++){
            for(int j=1;j<n;j++){
                if(matrix[0][j]==0||matrix[i][0]==0) matrix[i][j]=0;
            }
        }
        if(rowZero){
            for(int i=0;i<n;i++){
                matrix[0][i]=0;
            }
        }
        if(colZero){
            for(int i=0;i<m;i++){
                matrix[i][0]=0;
            }
        }
    }
};