頂端迭代器

給定一個迭代器類的介面，介面包含兩個方法： next() 和 hasNext()。設計並實現一個支援 peek() 操作的頂端迭代器 -- 其本質就是把原本應由 next() 方法返回的元素 peek() 出來。

示例:

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

呼叫 next() 返回 1，得到列表中的第一個元素。

現在呼叫 peek() 返回 2，下一個元素。在此之後呼叫 next() 仍然返回 2。

最後一次呼叫 next()

進階：你將如何拓展你的設計？使之變得通用化，從而適應所有的型別，而不只是整數型？

[思路]

設一個 peeked 的flag 同時 儲存 peeked 過得值.  如果已經peeked過了, next()直接返回儲存的值即可.

 1 // Java Iterator interface reference:
 2 // https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html
 3 class
 
 PeekingIterator implements Iterator<Integer> {
 4     Iterator<Integer> itr;
 5     boolean peeked;
 6     int peekVal;
 7 
 8     public PeekingIterator(Iterator<Integer> iterator) {
 9         // initialize any member here.
10         this.itr = iterator;
11     }
12 
13     // Returns the next element in the iteration without advancing the iterator.
 

14     public Integer peek() {
15         if(peeked) {
16             return peekVal;
17         } else {
18             peeked = true;
19             peekVal = itr.next();
20         }
21         return peekVal;
22     }
23 
24     // hasNext() and next() should behave the same as in the Iterator interface.
25     // Override them if needed.
26     @Override
27     public Integer next() {
28         if(peeked) {
29             peeked = false;
30             return peekVal;
31         } else return itr.next();
32     }
33 
34     @Override
35     public boolean hasNext() {
36         return peeked || itr.hasNext();
37     }
38 }