JAVA程序设计：顶端迭代器（LeetCode：284）

给定一个迭代器类的接口，接口包含两个方法： next() 和 hasNext()。设计并实现一个支持 peek() 操作的顶端迭代器 -- 其本质就是把原本应由 next() 方法返回的元素 peek() 出来。

示例:

假设迭代器被初始化为列表 [1,2,3]。

调用 next() 返回 1，得到列表中的第一个元素。

现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。

最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。

进阶：你将如何拓展你的设计？使之变得通用化，从而适应所有的类型，而不只是整数型？

思路：采用数组模拟即可，我们采用模拟动态扩容的方法进行优化。

class PeekingIterator implements Iterator<Integer> {

	private int size=8;
	private Integer[] elements=new Integer[size];
	private int p=0;
	private int q=p;
	
	public PeekingIterator(Iterator<Integer> iterator) {
	    // initialize any member here.
	    while(iterator.hasNext())
	    {
	    	if(q==size)
	    		grow();
	    	elements[q++]=iterator.next();
	    }
	}

    // Returns the next element in the iteration without advancing the iterator.
	public Integer peek() {
        return elements[p];
	}

	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	@Override
	public Integer next() {
		return elements[p++];
	}

	@Override
	public boolean hasNext() {
		return p!=q;
	}
	
	private void grow()
	{
		int newSize=size > Integer.MAX_VALUE >> 1 ? Integer.MAX_VALUE: size << 1;
		Integer[] objects=new Integer[newSize];
		System.arraycopy(elements, 0, objects, 0, size);
		size=newSize;
		elements=objects;
	}
}