给你一个嵌套的整型列表。请你设计一个迭代器,使其能够遍历这个整型列表中的所有整数。列表中的每一项或者为一个整数,或者是另一个列表。

示例 1:

输入: [[1,1],2,[1,1]]

输出: [1,1,2,1,1]

解释: 通过重复调用 next 直到 hasNext 返回 false,next 返回的元素的顺序应该是: [1,1,2,1,1]。

示例 2:

输入: [1,[4,[6]]]

输出: [1,4,6]

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

答案:

 1interface NestedInteger {
2    boolean isInteger();
3
4    Integer getInteger();
5
6    List<NestedInteger> getList();
7}
8
9class NestedIterator implements Iterator<Integer> {
10    Stack<NestedInteger> stack = new Stack<>();
11
12    public NestedIterator(List<NestedInteger> nestedList) {
13        for (int i = nestedList.size() - 1; i >= 0; i--) {
14            stack.push(nestedList.get(i));
15        }
16    }
17
18    @Override
19    public Integer next() {
20        return stack.pop().getInteger();
21    }
22
23    @Override
24    public boolean hasNext() {
25        while (!stack.isEmpty()) {
26            NestedInteger curr = stack.peek();
27            if (curr.isInteger()) {
28                return true;
29            }
30            stack.pop();
31            for (int i = curr.getList().size() - 1; i >= 0; i--) {
32                stack.push(curr.getList().get(i));
33            }
34        }
35        return false;
36    }
37}

解析:

使用一个栈来存储,在构造函数中压栈是从list的最后一个元素开始的,我们知道栈是先进后出的一种数据结构,所以最先压栈的肯定是最后出栈。这里的算法核心是在hasNext函数中,其中第27行判断NestedInteger中是否只有单独的一个元素,如果是的话就直接return true,否则把curr.getList中的元素还按照构造方法中的再一个个压栈。