Java實現 LeetCode 284 頂端迭代器

284. 頂端迭代器

給定一個迭代器類的接口，接口包含兩個方法： next() 和 hasNext()。設計並實現一個支持 peek() 操做的頂端迭代器 – 其本質就是把本來應由 next() 方法返回的元素 peek() 出來。

示例:

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

調用 next() 返回 1，獲得列表中的第一個元素。
如今調用 peek() 返回 2，下一個元素。在此以後調用 next() 仍然返回 2。
最後一次調用 next() 返回 3，末尾元素。在此以後調用 hasNext() 應該返回 false。
進階：你將如何拓展你的設計？使之變得通用化，從而適應全部的類型，而不僅是整數型？

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

private Iterator<Integer> iterator;
    private Integer cache = null; // 第一次peek時, 緩存迭代的元素

    public PeekingIterator(Iterator<Integer> iter) {
        iterator = iter;
    }

    public Integer peek() {
        if (cache == null)
            cache = iterator.next();
        return cache;
    }

    @Override
    public Integer next() {
        if (cache == null)
            return iterator.next();
        Integer temp = cache;
        cache = null;
        return temp;
    }

    @Override
    public boolean hasNext() {
        return cache != null || iterator.hasNext();
    }
}