Update Challenge.java

ByteLegendQuest · Nov 23, 2023 · 5401698 · 5401698
1 parent 6789560
commit 5401698
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];
     }
 }