每日一题 LeetCode 284. 窥探迭代器 java

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

class PeekingIterator implements Iterator<Integer> {
    Iterator<Integer> iterator;
    Integer nextElement=null;
	public PeekingIterator(Iterator<Integer> iterator) {
	    // initialize any member here.
	    this.iterator=iterator;
        this.nextElement=this.iterator.next();
	}
	
    // Returns the next element in the iteration without advancing the iterator.
	public Integer peek() {
        return nextElement;
	}
	
	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	@Override
	public Integer next() {
	    Integer res=nextElement;
        nextElement=iterator.hasNext()?iterator.next():null;
        return res;
	}
	
	@Override
	public boolean hasNext() {
	    return nextElement!=null;
	}
}