[N-0] refactor 341

fishercoder1534 · fishercoder1534 · commit 483dd5f53a5d · 2017-11-28T07:36:26.000-08:00
diff --git a/src/main/java/com/fishercoder/solutions/_341.java b/src/main/java/com/fishercoder/solutions/_341.java
@@ -11,30 +11,25 @@
  * 341. Flatten Nested List Iterator
  *
  * Given a nested list of integers, implement an iterator to flatten it.
-
- Each element is either an integer, or a list -- whose elements may also be integers or other lists.
-
- Example 1:
- Given the list [[1,1],2,[1,1]],
-
- By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,1,2,1,1].
-
- Example 2:
- Given the list [1,[4,[6]]],
-
- By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,4,6].
+ * Each element is either an integer, or a list -- whose elements may also be integers or other lists.
+ *
+ * Example 1:
+ * Given the list [[1,1],2,[1,1]],
+ * By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,1,2,1,1].
+ *
+ * Example 2:
+ * Given the list [1,[4,[6]]],
+ * By calling next repeatedly until hasNext returns false, the order of elements returned by next should be: [1,4,6].
  */
 public class _341 {
 
     public static class NestedIterator implements Iterator<Integer> {
 
         private Queue<Integer> flattenedList;
-        private Iterator<Integer> iterator;
 
         public NestedIterator(List<NestedInteger> nestedList) {
             flattenedList = new LinkedList<>();
             constructList(nestedList);
-            iterator = flattenedList.iterator();
         }
 
         private void constructList(List<NestedInteger> nestedList) {
@@ -49,12 +44,12 @@ private void constructList(List<NestedInteger> nestedList) {
 
         @Override
         public Integer next() {
-            return iterator.next();
+            return flattenedList.poll();
         }
 
         @Override
         public boolean hasNext() {
-            return iterator.hasNext();
+            return !flattenedList.isEmpty();
         }
     }
 
