array_setmatrix_zeroes2.cpp

#include<bits/stdc++.h>

using namespace std;

void set_zeroes(vector<vector<int>>&matrix,int n,int m){
    int col0=1;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(matrix[i][j]==0){
                matrix[i][0]=0;
                if(j!=0){
                    matrix[0][j]=0;
                }else{
                    col0=0;
                }
            }
        }
    }

    for(int i=1;i<n;i++){
        for(int j=1;j<m;j++){
            if(matrix[i][j]!=0){
                if(matrix[0][j]==0 || matrix[i][0]==0){
                    matrix[i][j]=0;
                }
            }
        }
    }
    if(matrix[0][0]==0){
        for(int j=0;j<m;j++) matrix[0][j]=0;
    }   
    if(col0==0){
        for(int i=0;i<n;i++) matrix[i][0]=0;
    }
}

int main(){
    vector<vector<int>>matrix={
        {1,1,1,1},
        {1,0,0,1},
        {1,1,0,1},
        {1,1,1,1}
    };
    for(auto it:matrix){
        for(auto itr: it) cout<<itr<<" ";
        cout<<endl;
    }
    cout<<endl;
    int n=matrix.size();
    int m=matrix[0].size();
    set_zeroes(matrix,n,m);
    for(auto it:matrix){
        for(auto itr: it) cout<<itr<<" ";
        cout<<endl;
    }

    return 0;
}