Time: 10 ms (71.64%) | Memory: 13.7 MB (76.12%) - LeetSync

Author: sayandip20

73-set-matrix-zeroes/set-matrix-zeroes.cpp

@@ -0,0 +1,42 @@
+class Solution {
+public:
+    void setZeroes(vector<vector<int>>& matrix) {
+        int n = matrix.size();
+        int m = matrix[0].size();
+        bool flag1 = false, flag2 = false;
+        for(int i=0; i<n; i++){
+            if(matrix[i][0] == 0){
+                flag1 = true;
+            }
+        }
+        for(int j=0; j<m; j++){
+            if(matrix[0][j] == 0){
+                flag2 = true;
+            }
+        }
+        for(int i=1; i<n; i++){
+            for(int j=1; j<m; j++){
+                if(matrix[i][j] == 0){
+                    matrix[i][0] = matrix[0][j] = 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(flag1 == true){
+            for(int i=0; i<n; i++){
+                matrix[i][0] = 0;
+            }
+        }
+        if(flag2 == true){
+            for(int j=0; j<m; j++){
+                matrix[0][j] = 0;
+            }
+        }
+    }
+};