Advertisement
vaibhav1906

Set Matrix to zero | Shweta Bandagale

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