Peeking Iterator【284】

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

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

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

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

	@Override
	public boolean hasNext() {
	    if(front != null){
	        return true;
	    }
	    return false;
	}
}

主要实现方式为双指针，按需返回内容

这里附上另外一份代码（其他网友的代码，仅供学习所用）

// Java Iterator interface reference:
// https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html
class PeekingIterator implements Iterator<Integer> {
    Integer peek = null;
    private Iterator<Integer> iterator;
	public PeekingIterator(Iterator<Integer> iterator) {
	    // initialize any member here.
	    this.iterator = iterator;
	}

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

	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	@Override
	public Integer next() {
	    if(peek != null) {
	        int temp = peek;
	        peek = null;
	        return temp;
	    }
	    return iterator.next();
	}

	@Override
	public boolean hasNext() {
	    if(peek != null) return true;
	    else return iterator.hasNext();
	}
}