leetcode Set Matrix Zeroes

不多说了。

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) 
    {
        int m=matrix.size();
        if(m==0)return;
        int n=matrix[0].size();
         vector<int>a;
        vector<int>b;       
        for(int i=0;i<m;i++)
        {
             for(int j=0;j<n;j++)
             {
                 if(matrix[i][j]==0)
                 {
                     a.push_back(i);
                     b.push_back(j);
                 }
             }        
        }
        for(int i=0;i<a.size();i++)
        {
            for(int j=0;j<n;j++)
            {
                matrix[a[i]][j]=0;
            }
        }
        for(int i=0;i<b.size();i++)
        {
            for(int j=0;j<m;j++)
            {
                matrix[j][b[i]]=0;
            }
        }       
    }
};