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

2023-12-13 12:18:54

力扣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);
        }

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

附题目数据类型:

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

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

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

文章来源:https://blog.csdn.net/qq_43511039/article/details/134965288
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。