Leetcode刷题341. 扁平化嵌套列表迭代器

给你一个嵌套的整数列表 nestedList 。每个元素要么是一个整数，要么是一个列表；该列表的元素也可能是整数或者是其他列表。请你实现一个迭代器将其扁平化，使之能够遍历这个列表中的所有整数。

实现扁平迭代器类 NestedIterator ：

NestedIterator(List<NestedInteger> nestedList) 用嵌套列表 nestedList 初始化迭代器。
int next() 返回嵌套列表的下一个整数。
boolean hasNext() 如果仍然存在待迭代的整数，返回 true ；否则，返回 false 。
你的代码将会用下述伪代码检测：

initialize iterator with nestedList
res = []
while iterator.hasNext()
    append iterator.next() to the end of res
return res
如果 res 与预期的扁平化列表匹配，那么你的代码将会被判为正确。

示例 1：

输入：nestedList = [[1,1],2,[1,1]]
输出：[1,1,2,1,1]
解释：通过重复调用 next 直到 hasNext 返回 false，next 返回的元素的顺序应该是: [1,1,2,1,1]。
示例 2：

输入：nestedList = [1,[4,[6]]]
输出：[1,4,6]
解释：通过重复调用 next 直到 hasNext 返回 false，next 返回的元素的顺序应该是: [1,4,6]。
 

提示：

1 <= nestedList.length <= 500
嵌套列表中的整数值在范围 [-106, 106] 内

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/flatten-nested-list-iterator
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

感谢labuladong大佬的详细解法树的遍历框架：题目不让我做什么，我偏要做

方法一：嵌套的整型列表是一个树形结构，树上的叶子节点对应一个整数，非叶节点对应一个列表。先遍历整个嵌套列表，将所有整数存入一个数组，然后遍历该数组从而实现next和hasNext方法。
时间和空间复杂度O(N)

public class NestedIterator implements Iterator<Integer> {

    private Iterator<Integer> it;
    public NestedIterator(List<NestedInteger> nestedList) {
        List<Integer> result = new LinkedList<>();
        for (NestedInteger node : nestedList) {
            traverse(node, result);
        }
        it = result.iterator();
    }

    private void traverse(NestedInteger node, List<Integer> list) {
        //遇到叶子节点时添加结果
        if (node.isInteger()) {
            list.add(node.getInteger());
            return;
        }
        for (NestedInteger child : node.getList()) {
            traverse(child, list);
        }
    }

    @Override
    public Integer next() {
        return it.next();
    }

    @Override
    public boolean hasNext() {
        return it.hasNext();
    }
}

 方法二：方法一如果输入的规模非常大，构造函数中的计算就会很慢，而且很占用内存。
调用hasNext时如果第一个元素是列表类型，则不断展开整个元素，直到第一个元素是整数

时间复杂度：初始化和next为O(1)，hasNext为均摊O(1)
空间复杂度：O(N)

public class NestedIterator implements Iterator<Integer> {
    
    private LinkedList<NestedInteger> list;
    public NestedIterator(List<NestedInteger> nestedList) {
        list = new LinkedList<>(nestedList);
    }

    @Override
    public Integer next() {
        //hasNext方法保证了第一个元素一定是整数类型
        return list.remove(0).getInteger();
    }

    @Override
    public boolean hasNext() {
        while (!list.isEmpty() && !list.get(0).isInteger()) {
            //不断展开，直到列表第一个元素是整数类型
            List<NestedInteger> first = list.remove(0).getList();
            //这里为什么要逆序遍历？使元素的顺序保持一致
            for (int i = first.size() - 1; i >= 0; i--) {
                list.addFirst(first.get(i));
            }
        }
        return !list.isEmpty();
    }
}