Add c++ implementation

Signed-off-by: begeekmyfriend <[email protected]>

Author: begeekmyfriend

0073_set_matrix_zeroes/set_zero.c

@@ -1,36 +1,36 @@
 #include <stdio.h>
 #include <stdlib.h>
 
-static void setZeroes(int** matrix, int matrixRowSize, int matrixColSize)
+
+void setZeroes(int** matrix, int matrixSize, int* matrixColSize)
 {
-    int row, col, bRow = 0, bCol = 0;
-    for (row = 0; row < matrixRowSize; row++) {
-        for (col = 0; col < matrixColSize; col++) {
+    int row, col;
+    bool bRow = false, bCol = false;
+    for (row = 0; row < matrixSize; row++) {
+        for (col = 0; col < matrixColSize[row]; col++) {
             if (matrix[row][col] == 0) {
-                if (row == 0) bCol = 1;
-                if (col == 0) bRow = 1;
+                if (row == 0) bRow = true;
+                if (col == 0) bCol = true;
                 matrix[0][col] = matrix[row][0] = 0;
             }
         }
     }
 
-    for (row = 1; row < matrixRowSize; row++) {
-        for(col = 1; col < matrixColSize; col++){
+    for (row = 1; row < matrixSize; row++) {
+        for(col = 1; col < matrixColSize[row]; col++){
             if (matrix[0][col] == 0 || matrix[row][0] == 0) {
                 matrix[row][col] = 0;
             }
         }
     }
 
     if (bRow) {
-        for(row = 0; row < matrixRowSize; row++) {
-            matrix[row][0] = 0;
-        }
+        memset(matrix[0], 0, matrixColSize[0] * sizeof(int));
     }
 
     if (bCol) {
-        for (col = 0; col <matrixColSize; col++) {
-            matrix[0][col] = 0;
+        for (row = 0; row < matrixSize; row++) {
+            matrix[row][0] = 0;
         }
     }
 }

0073_set_matrix_zeroes/set_zero.cc

@@ -0,0 +1,38 @@
+#include <bits/stdc++.h>
+
+using namespace std;
+
+public:
+    void setZeroes(vector<vector<int>>& matrix) {
+        bool bRow = false, bCol = false;
+        for (int row = 0; row < matrix.size(); row++) {
+            for (int col = 0; col < matrix[row].size(); col++) {
+                if (matrix[row][col] == 0) {
+                    if (row == 0) { bRow = true; }
+                    if (col == 0) { bCol = true; }
+                    matrix[0][col] = matrix[row][0] = 0;
+                }
+            }
+        }
+
+        for (int row = 1; row < matrix.size(); row++) {
+            for (int col = 1; col < matrix[row].size(); col++) {
+                if (matrix[0][col] == 0 || matrix[row][0] == 0) {
+                    matrix[row][col] = 0;
+                }
+            }
+        }
+
+        if (bRow) {
+            for (auto& m : matrix[0]) {
+                m = 0;
+            }
+        }
+
+        if (bCol) {
+            for (int row = 0; row < matrix.size(); row++) {
+                matrix[row][0] = 0;
+            }
+        }
+    }
+};