setmatrixzero.cpp

 #include<bits/stdc++.h>
using namespace std;
 
 void setZeros(vector<vector<int>> &matrix)

{

// Write your code here.

  bool col = false, row = false;

   

   int n = matrix.size(), m = matrix[0].size();

   

   for(int i = 0 ; i < n ; i++)

   {

       for(int j = 0 ; j < m ; j++)

       {

           if(matrix[i][j] == 0)

           {

               if(j == 0) col = true;

               if(i == 0) row = true;    

               matrix[0][j] = 0;

               matrix[i][0] = 0;

           }

       }

   }

   

   for(int i = 1 ; i < n ; i++)

   {

       for(int j = 1 ; j < m ; j++)

       {

           if(matrix[i][0] == 0 || matrix[0][j] == 0) matrix[i][j] = 0;

       }

   }

   

   if(col){

       for(int i = 0 ; i < n ; i++) matrix[i][0] = 0;

   }

   if(row){

       for(int i = 0 ; i < m ; i++) matrix[0][i] = 0;

   }

   

}