面试题：给定一个迭代器类的接口，接口包含两个方法： next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器

给定一个迭代器类的接口，接口包含两个方法： next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器 -- 其本质就是把原本应由 next() 方法返回的元素 peek() 出来。

示例:

假设迭代器被初始化为列表 [1,2,3]。

调用 next() 返回 1，得到列表中的第一个元素。
现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。
最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。
进阶：你将如何拓展你的设计？使之变得通用化，从而适应所有的类型，而不只是整数型？

next()和hasNext()函数不必多说了，感觉这道题的重点在于考察如何实现这个peek()函数。首先还是创建一个迭代器开始迭代list，同时需要创建一个变量next来记录下一个会被迭代到的数字。

初始化的时候，如果迭代器有下一个元素hasNext()，那么就直接把这个元素拿到，放入next变量中暂时记录下来。这样peek()的时候，直接return这个next就行了。

hasNext()函数则是通过判断这个next变量是否为空，来判断是否整个list已经被遍历完了。

最后重点在于如何处理next()函数。因为next已经记录了下一个应该会被返回的值，所以返回这个数字不难，但是此时的next()函数，应该做如下这两步

• res = next - 把这个next数字拿过来，然后返回res，这样next()就跟peek()返回的是一样的数字了
• 如果迭代器还有下一个元素，则把这个元素拿出来，用next变量记住，否则就返回NULL

在迭代器不为空的情况下，peek()和next()需要返回的是相同的东西，所以这里重写的next()函数实际上是在看再下一个元素。

时间O(n)

空间O(1)

Java实现

// Java Iterator interface reference:
// https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html
class PeekingIterator implements Iterator<Integer> {
    private Iterator<Integer> iter;
    private Integer next = null;

    public PeekingIterator(Iterator<Integer> iterator) {
        // initialize any member here.
        iter = iterator;
        if (iter.hasNext()) {
            next = iter.next();
        }
    }

    // Returns the next element in the iteration without advancing the iterator.
    public Integer peek() {
        return next;
    }

    // hasNext() and next() should behave the same as in the Iterator interface.
    // Override them if needed.
    @Override
    public Integer next() {
        Integer res = next;
        next = iter.hasNext() ? iter.next() : null;
        return res;
    }

    @Override
    public boolean hasNext() {
        return next != null;
    }
}