力扣labuladong一刷day38天扁平化嵌套列表迭代器

当年拼却醉颜红2023-12-14 17:58

力扣labuladong一刷day38天扁平化嵌套列表迭代器

一、341. 扁平化嵌套列表迭代器

题目链接：https://leetcode.cn/problems/flatten-nested-list-iterator/description/

思路：本题是一个嵌套列表，如果一次性全部给拉平迭代到一维占用的内存可能过大。我们采用惰性加载，因为使用的时候是先判断hasNext()然后才调用next，每次当判断hasNext()的时候，如果list中第一个元素是list那么就惰性加载给展开，展开的时候就把当前局部的list倒序添加都原list索引为0的位置。

java 复制代码

 public class NestedIterator implements Iterator<Integer> {

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

        @Override
        public Integer next() {
            return list.remove(0).getInteger();
        }

        @Override
        public 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();
        }
    }

附题目数据类型：

java 复制代码

public class NestedInteger {
    // 如果其中存的是一个整数，则返回 true，否则返回 false
    public boolean isInteger();

    // 如果其中存的是一个整数，则返回这个整数，否则返回 null
    public Integer getInteger();

    // 如果其中存的是一个列表，则返回这个列表，否则返回 null
    public List<NestedInteger> getList();
}