给你一个嵌套的整型列表。请你设计一个迭代器,使其能够遍历这个整型列表中的所有整数。列表中的每一项或者为一个整数,或者是另一个列表。
示例 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中的元素还按照构造方法中的再一个个压栈。