[Hacker Rank]: Warmup. Staircase solved ✅.

Author: Gonzalo Diaz

src/lib/exercises/include/exercises/hackerrank/warmup/staircase.h

@@ -6,6 +6,7 @@ extern "C" {
 
 char **HACKERRANK_WARMUP_staircaseCalculate(int n);
 void HACKERRANK_WARMUP_staircase(int n);
+void HACKERRANK_WARMUP_freeStaircase(char **staircase, int n);
 
 #ifdef __cplusplus
 } // extern "C"

src/lib/exercises/src/hackerrank/warmup/staircase.c

@@ -29,13 +29,19 @@ char **HACKERRANK_WARMUP_staircaseCalculate(int n) {
   return answer;
 }
 
+void HACKERRANK_WARMUP_freeStaircase(char **staircase, int n) {
+  for (int i = 0; i < n; i++) {
+    free(staircase[i]);
+  }
+  free(staircase);
+}
+
 void HACKERRANK_WARMUP_staircase(int n) {
   char **output = HACKERRANK_WARMUP_staircaseCalculate(n);
 
   for (int i = 0; i < n; i++) {
     printf("%s\n", output[i]);
-    free(output[i]);
   }
 
-  free(output);
+  HACKERRANK_WARMUP_freeStaircase(output, n);
 }

src/tests/unit/lib/hackerrank/warmup/staircase.test.cpp

@@ -29,6 +29,8 @@ TEST_CASE("staircase", "[warmup]") {
       result_as_vector.emplace_back(result[i]);
     }
 
+    HACKERRANK_WARMUP_freeStaircase(result, input);
+
     CHECK(result_as_vector == testcase["expected"]);
 
     // Just call void function, to collect coverage