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.

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?

思路一：O(m+n)

首先实现了 空间 O(m+n) 的算法

代码一

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
       int row = matrix.size();
       if(row<=0)
          return ;
       vector<int> zeroRows;
       vector<int> zeroCols;
       int col = matrix[0].size();
       for(int i=0;i<row;i++)
           for(int j=0;j<col;j++)
           {
               if(matrix[i][j]==0)
               {
                   zeroRows.push_back(i);
                   break;
               }
           }
       for(int i=0;i<col;i++)
           for(int j=0;j<row;j++)
           {
               if(matrix[j][i]==0)
               {
                   zeroCols.push_back(i);
                   break;
               }
           }
       for(int i=0;i<zeroRows.size();i++)
       {
           int r = zeroRows[i];
           for(int j=0;j<col;j++)
               matrix[r][j]=0;    
       }
       for(int i=0;i<zeroCols.size();i++)
       {
           int c = zeroCols[i];
           for(int j=0;j<row;j++)
               matrix[j][c]=0;    
       }
       
    }
};

实现二：没有空间

通过第一行和第一列来记录矩阵中的所有含有0的行和列。

代码二

class Solution {
public:
    void setZeroes(vector<vector<int> > &matrix) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
       int row = matrix.size();
       if(row<=0)
          return ;
       int col = matrix[0].size();
       bool firstRow = false;
       bool firstCol = false;
       for(int i=0;i<col;i++)
           if(matrix[0][i]==0)
           {
               firstRow = true;
               break;
           }
       for(int i=0;i<row;i++)
           if(matrix[i][0]==0)
           {
               firstCol = 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<row;i++)
           for(int j=1;j<col;j++)
           {
               if(matrix[i][0]==0 || matrix[0][j]==0)
                   matrix[i][j]=0;
           }
       if(firstRow)
       {
           for(int j=0;j<col;j++)
               matrix[0][j]=0;    
       }
       if(firstCol)
       {
           for(int j=0;j<row;j++)
               matrix[j][0]=0;    
       }
       
    }
};