fix(codes/c): Fix unknown behavior when size is 0 or arr is a null po…

…inter (krahets#288) Co-authored-by: Yudong Jin <[email protected]>
williamherry · Jan 20, 2023 · 0b5761e · 0b5761e
1 parent 31eefe0
commit 0b5761e
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/codes/c/include/print_util.h b/codes/c/include/print_util.h
@@ -27,17 +27,21 @@ extern "C" {
  */
 static void printArray(int arr[], int size) {
     printf("[");
-    for (int i = 0; i < size - 1; i++) {
-        if (arr[i] != NIL) {
-            printf("%d, ", arr[i]);
-        } else {
-            printf("NULL, ");
-        }
-    }
-    if (arr[size - 1] != NIL) {
-        printf("%d]\n", arr[size - 1]);
+    if (arr != NULL && size != 0) {
+      for (int i = 0; i < size - 1; i++) {
+          if (arr[i] != NIL) {
+              printf("%d, ", arr[i]);
+          } else {
+              printf("NULL, ");
+          }
+      }
+      if (arr[size - 1] != NIL) {
+          printf("%d]\n", arr[size - 1]);
+      } else {
+          printf("NULL]\n");
+      }
     } else {
-        printf("NULL]\n");
+      printf("]");
     }
 }