[LeetCode]341. Flatten Nested List Iterator

gqk289

于 2017-03-27 11:24:56 发布

阅读量268

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/gqk289/article/details/66969682

版权

LeetCode 专栏收录该内容

324 篇文章

订阅专栏

本文介绍了一种解决LeetCode上'Flatten Nested List Iterator'问题的方法。通过使用栈结构来遍历嵌套列表，确保每次都能从栈顶取出Integer类型的元素。实现了NestedIterator类的构造方法、next方法及hasNext方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://leetcode.com/problems/flatten-nested-list-iterator/#/description

迭代器都是先调用hasNext再调用next，因此保证stack最顶上的一个是Integer即可

public class NestedIterator implements Iterator<Integer> {
    
    Stack<NestedInteger> stack;

    public NestedIterator(List<NestedInteger> nestedList) {
        stack = new Stack();
        for (int i = nestedList.size() - 1; i >= 0; i--) {
            stack.push(nestedList.get(i));
        }
    }

    @Override
    public Integer next() {
        return stack.pop().getInteger();
    }

    @Override
    public boolean hasNext() {
        while (!stack.isEmpty()) {
            NestedInteger nest = stack.peek();
            if (nest.isInteger()) {
                return true;
            }
            nest = stack.pop();
            for (int i = nest.getList().size() - 1; i >= 0; i--) {
                stack.push(nest.getList().get(i));
            }
        }
        return false;
    }
}