updating method signature in flatten array

Author: jagdish-15

exercises/practice/flatten-array/.meta/config.json

@@ -6,6 +6,7 @@
     "aadityakulkarni",
     "FridaTveit",
     "jackattack24",
+    "jagdish-15",
     "jmrunkle",
     "jtigger",
     "kytrinyx",

exercises/practice/flatten-array/.meta/src/reference/java/Flattener.java

@@ -1,28 +1,17 @@
 import java.util.ArrayList;
-import java.util.Collections;
 import java.util.List;
 
-final class Flattener {
+class Flattener {
 
-    List flatten(final List nestedList) {
-        if (nestedList.isEmpty()) {
-            return new ArrayList<>();
-        } else {
-            final List result = new ArrayList();
-
-            final Object head = nestedList.get(0);
-            final List tail = nestedList.subList(1, nestedList.size());
-
-            if (head instanceof List) {
-                result.addAll(flatten((List) head));
-            } else {
-                result.add(head);
+    List<Object> flatten(List<?> list) {
+        List<Object> flattenedList = new ArrayList<>();
+        for (Object element: list) {
+            if (element instanceof List<?> listAsElement) {
+                flattenedList.addAll(flatten(listAsElement));
+            } else if (element != null) {
+                flattenedList.add(element);
             }
-
-            result.addAll(flatten(tail));
-            result.removeAll(Collections.singleton(null));
-            return result;
         }
+        return flattenedList;
     }
-
 }

exercises/practice/flatten-array/src/main/java/Flattener.java

@@ -2,7 +2,7 @@
 
 class Flattener {
 
-    <T> List<T> flatten(List<T> list) {
+    List<Object> flatten(List<?> list) {
         throw new UnsupportedOperationException("Delete this statement and write your own implementation.");
     }