给定一个迭代器类的接口,接口包含两个方法: next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器 -- 其本质就是把原本应由 next() 方法返回的元素 peek() 出来。
示例:
假设迭代器被初始化为列表 [1,2,3]。
调用 next() 返回 1,得到列表中的第一个元素。
现在调用 peek() 返回 2,下一个元素。在此之后调用 next() 仍然返回 2。
最后一次调用 next() 返回 3,末尾元素。在此之后调用 hasNext() 应该返回 false。
class PeekingIterator implements Iterator<Integer> {
Queue<Integer> queue=new LinkedList();
public PeekingIterator(Iterator<Integer> iterator) {
while(iterator.hasNext()){
queue.add(iterator.next());
}
}
public Integer peek() {
return queue.peek();
}
// Override them if needed.
@Override
public Integer next() {
return queue.remove();
}
@Override
public boolean hasNext() {
if(queue.size()>0){
return true;
}else{
return false;
}
}
}