knakul853

Untitled

Jul 11th, 2020
143
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.86 KB | None | 0 0
  1. /**
  2. knakul853
  3. **/
  4. class Solution {
  5. public:
  6.     void setZeroes(vector<vector<int>>& matrix) {
  7.        
  8.         if(matrix.empty()) return;
  9.        
  10.         int n = (int)matrix.size();
  11.         int m = (int)matrix[0].size();
  12.        
  13.         int col0 = 1;
  14.        
  15.         for(int i=0;i<n;i++)
  16.         {
  17.             if(matrix[i][0] == 0) col0 = 0;
  18.             for(int j=1; j<m; j++)
  19.             {
  20.                 if(matrix[i][j] == 0 )
  21.                     matrix[i][0] = matrix[0][j] = 0;
  22.             }
  23.         }
  24.        
  25.         for(int i=n-1;i>=0;i--)
  26.         {
  27.             for(int j=m-1;j>=1;j--)
  28.             {
  29.                 if(matrix[i][0]==0 || matrix[0][j] == 0 )
  30.                 {
  31.                     matrix[i][j] = 0;
  32.                 }
  33.             }
  34.            
  35.             if(col0 == 0)matrix[i][0] = 0;
  36.         }
  37.        
  38.     }
  39. };
Add Comment
Please, Sign In to add comment