4/06/2014

Leetcode -- Set Matrix Zeroes

class Solution {  
 public:  
   void setZeroes(vector<vector<int> > &matrix) {  
     vector<vector<int> > zeros;  
     for(int i = 0; i < matrix.size(); ++i) {  
       for(int j = 0; j < matrix[0].size(); ++j) {  
         if(matrix[i][j] == 0) {  
           vector<int> ax{i, j};  
           zeros.emplace_back(ax);  
         }  
       }  
     }  
     for(auto& v: zeros) {  
       for(int j = 0; j <matrix[0].size(); ++j) {  
         matrix[v[0]][j] = 0;  
       }  
       for(int i = 0; i < matrix.size(); ++i) {  
         matrix[i][v[1]] = 0;  
       }  
     }  
   }  
 };  

No comments: