力扣labuladong一刷day38天扁平化嵌套列表迭代器
一、341. 扁平化嵌套列表迭代器
题目链接:https://leetcode.cn/problems/flatten-nested-list-iterator/description/
思路:本题是一个嵌套列表,如果一次性全部给拉平迭代到一维占用的内存可能过大。我们采用惰性加载,因为使用的时候是先判断hasNext()然后才调用next,每次当判断hasNext()的时候,如果list中第一个元素是list那么就惰性加载给展开,展开的时候就把当前局部的list倒序添加都原list索引为0的位置。
public class NestedIterator implements Iterator<Integer> {private LinkedList<NestedInteger> list;public NestedIterator(List<NestedInteger> nestedList) {list = new LinkedList<>(nestedList);}@Overridepublic Integer next() {return list.remove(0).getInteger();}@Overridepublic boolean hasNext() {while (!list.isEmpty() && !list.get(0).isInteger()) {List<NestedInteger> nList = list.remove(0).getList();for (int i = nList.size()-1; i >= 0; i--) {list.addFirst(nList.get(i));}}return !list.isEmpty();}}
附题目数据类型:
public class NestedInteger {// 如果其中存的是一个整数,则返回 true,否则返回 falsepublic boolean isInteger();// 如果其中存的是一个整数,则返回这个整数,否则返回 nullpublic Integer getInteger();// 如果其中存的是一个列表,则返回这个列表,否则返回 nullpublic List<NestedInteger> getList();
}