[Leetcode] 284. Peeking Iterator

最新推荐文章于 2024-08-01 07:32:18 发布

原创最新推荐文章于 2024-08-01 07:32:18 发布 · 133 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

294 篇文章

订阅专栏

本文介绍了一种高效实现PeekingIterator的方法，通过预取iterator的下一个元素并缓存，使得peek()和next()操作都能在O(1)时间内完成。此方法适用于任何实现了Iterator接口的数据结构。

这一题其实只要找对了方法就是异常简单的一题，我觉得这题其实应该归类为easy。但是既然LEETCODE写了medium我就记录一下吧。

这题做法就是一个，先把iterator next的值保留好，如果call peek，那么就直接返回保留好的值，如果call next，就返回保留好的值的同时，从iterator再取一个next回来。所以实际上iterator当前的状态是已经没有next的，next的值我们已经拿出来放在一个地方了。根据这个理论，很快就可以得到下面这个答案

class PeekingIterator implements Iterator<Integer> {

    Integer current = null;
    Iterator<Integer> iterator;
	public PeekingIterator(Iterator<Integer> iterator) {
	    // initialize any member here.
	    this.current = iterator.hasNext() ? iterator.next() : null;
        this.iterator = iterator;
	}

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

	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	@Override
	public Integer next() {
	    Integer result = this.current;
        this.current = iterator.hasNext() ? iterator.next() : null;
        return result;
	}

	@Override
	public boolean hasNext() {
	    return this.current != null;
	}
}

Follow up的话其实应该指的是java generics，可以看看这个：http://tutorials.jenkov.com/java-generics/implementing-iterable.html或者https://docs.oracle.com/javase/tutorial/java/generics/types.html或者https://docs.oracle.com/javase/tutorial/java/generics/types.html