LeetCode - 284. Peeking Iterator

来源:互联网 发布:软件系统调试 编辑:程序博客网 时间:2024/06/05 17:44

在已有的类的基础上实现功能更为强大的一个子类,这时我们可以把父类作为子类的一个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.htmlclass 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.@Overridepublic Integer next() {    Integer result = next;    next = iter.hasNext()? iter.next() : null;    return result;}@Overridepublic boolean hasNext() {    return next != null;}}

思考:

1. iterator类的题目通常会是要求实现next(), hasNext(), remove()等方法

2. iterator类的问题通常需要使用一个指针指向下一个需要返回的元素,methods的实现都基于此

3. next()方法通常需要两个步骤,第一个步骤是找出需要返回的元素,而下一步则是将指针重新设置到下一个需要返回的元素上面

0 0