LeetCode/problems/peeking-iterator.java

33 lines
588 B
Java
Raw Normal View History

class PeekingIterator implements Iterator<Integer> {
private Iterator<Integer> it;
private Integer pVal;
private boolean pHasNext;
public PeekingIterator(Iterator<Integer> iterator) {
it = iterator;
next();
}
public Integer peek() {
return pVal;
}
@Override
public Integer next() {
pHasNext = it.hasNext();
Integer oldValue = pVal;
if (pHasNext) {
pVal = it.next();
}
return oldValue;
}
@Override
public boolean hasNext() {
return pHasNext;
}
}