LeetCode284 窥探迭代器

Posted 2021-10-24 So istes immer

目录 

题目

请你设计一个迭代器，除了支持 hasNext 和 next 操作外，还支持 peek 操作。

实现 PeekingIterator 类：

PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素，并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素，返回 true ；否则，返回 false 。
int peek() 返回数组中的下一个元素，但 不 移动指针。

分析 

// Java Iterator interface reference:
// https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html

class PeekingIterator implements Iterator<Integer> {
	Iterator<Integer> iter;
    Integer next;
    public PeekingIterator(Iterator<Integer> iterator) {
        iter = iterator;
        next = iter.next();
	}
	
	public Integer peek() {
        return next;
	}
	
	public Integer next() {
	    Integer ans = next;
        next = iter.hasNext() ? iter.next() : null;
        return ans;
	}
	
	public boolean hasNext() {
	    return next != null;
	}
}