Update Challenge.java (#125)

Update Challenge.java (#125) Co-authored-by: VacancyIF <[email protected]>
ByteLegendQuest · Nov 23, 2023 · 22c858a · 22c858a
2 parents ea0afeb + 60b665e
commit 22c858a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/main/java/com/bytelegend/Challenge.java b/src/main/java/com/bytelegend/Challenge.java
@@ -11,10 +11,16 @@ public static void main(String[] args) {
     }
 
     public static Object firstOrNull(Object[] array) {
-        return array[1];
+        if(array.length == 0) {
+            return null;
+        }
+        return array[0];
     }
 
     public static Object lastOrNull(Object[] array) {
-        return array[array.length];
+        if(array.length == 0) {
+            return null;
+        }
+        return array[array.length - 1];
     }
 }