Set Matrix Zeroes

class Solution {public:    void setZeroes(vector<vector<int> > &matrix) {        int m=matrix.size();        int n=matrix[0].size();        vector<int>row;        vector<int>col;        row.assign(m,0);        col.assign(n,0);        for(int i=0;i<m;i++)        {            for(int j=0;j<n;j++)            {                if(matrix[i][j]==0)                {                    row[i]=1;                    col[j]=1;                }            }        }        for(int i=0;i<m;i++)        {            if(row[i]==1)               matrix[i].assign(n,0);        }            for(int j=0;j<n;j++)            {                if(col[j]==1)                   for(int k=0;k<m;k++)                       matrix[k][j]=0;            }            }};

感觉写得还是有点乱，主要是下面赋值部分，这里的空间复杂度是o(n+m)，题目中有问到是否能用常数的空间复杂度，可以考虑将用matrix的第一行，第一列作为上面row、col的替代，就是说：找到一个matrix[i][j]=0就可以将对应在第一行第一列的值赋值为0，最后再扫描一遍即可