lintcode-矩阵归零-162

给定一个m×n矩阵，如果一个元素是0，则将其所在行和列全部元素变成0。

需要在原地完成。

样例

给出一个矩阵[[1,2],[0,3]]，返回[[0,2],[0,0]]

挑战

你是否使用了额外的空间？

一个直接的解决方案是使用O(MN)的额外空间，但这并不是一个好的方案。

一个简单的改进方案是使用O(M + N)的额外空间，但这仍然不是最好的解决方案。

你能想出一个常数空间的解决方案吗？

O(M*N)

class Solution {
public:
 
    void setZeroes(vector<vector<int> > &matrix) {
        if(matrix.empty())
            return ;
        const int row=matrix.size();
        const int cal=matrix[0].size();
        
        int Row[row],Cal[cal];  //记录哪行哪列有零
        memset(Row,0,sizeof(Row));
        memset(Cal,0,sizeof(Cal));
        
        for(int i=0;i<row;++i)
            for(int j=0;j<cal;++j)
                if(0==matrix[i][j]){
                    Row[i]=1;
                    Cal[j]=1;
                }
                      
        for(int i=0;i<row;++i)    //置零操作，把两部分合成一部分了
            for(int j=0;j<cal;++j)
                if(Row[i]||Cal[j])
                    matrix[i][j]=0;
        
    }
};