From 1fab3fa20422d14d68930b1b2fc45ebafe9e001e Mon Sep 17 00:00:00 2001
From: YuCheng Hu <yuchenghu@ossez.com>
Date: Thu, 27 Dec 2018 13:49:22 -0500
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=B8=80=E4=B8=AA=20Java=208?=
 =?UTF-8?q?=20=E7=9A=84=E6=96=B9=E6=B3=95=E5=92=8C=E4=B8=80=E4=B8=AA?=
 =?UTF-8?q?=E9=80=92=E5=BD=92=E7=9A=84=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../interview/tests/PillPackTest.java         | 33 ++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/src/test/java/com/ossez/codebank/interview/tests/PillPackTest.java b/src/test/java/com/ossez/codebank/interview/tests/PillPackTest.java
index 1c7d284dea..688a12dea4 100644
--- a/src/test/java/com/ossez/codebank/interview/tests/PillPackTest.java
+++ b/src/test/java/com/ossez/codebank/interview/tests/PillPackTest.java
@@ -40,8 +40,37 @@ public class PillPackTest {
 		logger.debug("Test FlattenNestedArrays");
 
 		Object[] array = { 1, 2, new Object[] { 3, 4, new Object[] { 5, new Object[] { new Object[] { 6 } } }, 7 }, 8, 9, 10 };
-		logger.debug("{} - > {}", Arrays.deepToString(array), Arrays.toString(java8Flatten(array).toArray()));
+		logger.debug("LOOP: {} - > {}", Arrays.deepToString(array), Arrays.toString(loopFlatten(array)));
 
+		logger.debug("Java 8: {} - > {}", Arrays.deepToString(array), Arrays.toString(java8Flatten(array).toArray()));
+
+	}
+
+	/**
+	 * Loop And Recursive
+	 * 
+	 * @param inputArray
+	 * @return
+	 * @throws IllegalArgumentException
+	 */
+	private static Integer[] loopFlatten(Object[] inputArray) throws IllegalArgumentException {
+		// NULL CHECK
+		if (inputArray == null)
+			return null;
+
+		List<Integer> flatList = new ArrayList<Integer>();
+
+		for (Object element : inputArray) {
+			if (element instanceof Integer) {
+				flatList.add((Integer) element);
+			} else if (element instanceof Object[]) {
+				// Recursive
+				flatList.addAll(Arrays.asList(loopFlatten((Object[]) element)));
+			} else {
+				throw new IllegalArgumentException("Input must be an array of Integers or nested arrays of Integers");
+			}
+		}
+		return flatList.toArray(new Integer[flatList.size()]);
 	}
 
 	/**
@@ -51,7 +80,9 @@ public class PillPackTest {
 	 * @return
 	 */
 	private static Stream<Object> java8Flatten(Object[] array) {
+		// int[] flatInt = java8Flatten(array).mapToInt(Integer.class::cast).toArray();
 		return Arrays.stream(array).flatMap(o -> o instanceof Object[] ? java8Flatten((Object[]) o) : Stream.of(o));
+
 	}
 
 }