在已有的类的基础上实现功能更为强大的一个子类,这时我们可以把父类作为子类的一个attribute,通过这个attribute来使用父类的一些methods(对于这道题目是implements interface)。
既然这道题目要求实现的方法都与下一个有关,我们可以使用一个指针指向下一个需要返回的元素。对于constructor,我们只要对类里的变量进行初始化,将next指针指向下一个需要返回的元素即可;对于hasNext(),如果next指针是null的话,那么返回null,而如果next指针不为空的话,说明我们有下一个元素,所以这个方法的实现只要测试next指针是否为空即可。
而next()的实现要相对复杂一些,其实对于所有iterator的next方法的实现都要分为两步,第一步是得到需要返回的元素,第二步就是将当前的指针再次指向下一个需要返回的元素,为下一次next()的调用做准备。注意因为我们是在这个函数中调用了接口的方法hasNext(),注意这种将新实现的方法建立在原来方法的基础之上的思想。整体的代码如下:
// Java Iterator interface reference:
// https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html
class PeekingIterator implements Iterator<Integer> {
private Integer next = null;
private Iterator<Integer> iter;
public PeekingIterator(Iterator<Integer> iterator) {
// initialize any member here.
iter = iterator;
if(iter.hasNext()){
next = iter.next();
}
}
// Returns the next element in the iteration without advancing the iterator.
public Integer peek() {
return next;
}
// hasNext() and next() should behave the same as in the Iterator interface.
// Override them if needed.
@Override
public Integer next() {
Integer result = next;
next = iter.hasNext()? iter.next() : null;
return result;
}
@Override
public boolean hasNext() {
return next != null;
}
}
思考:
1. iterator类的题目通常会是要求实现next(), hasNext(), remove()等方法
2. iterator类的问题通常需要使用一个指针指向下一个需要返回的元素,methods的实现都基于此
3. next()方法通常需要两个步骤,第一个步骤是找出需要返回的元素,而下一步则是将指针重新设置到下一个需要返回的元素上面