Leetcode 284.顶端迭代器

Posted kexinxin

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Leetcode 284.顶端迭代器相关的知识,希望对你有一定的参考价值。

顶端迭代器

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

示例:

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

 

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

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

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

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

 

[思路]

设一个 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 }

 

以上是关于Leetcode 284.顶端迭代器的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode 284 窥探迭代器[迭代器 构造类] HERODING的LeetCode之路

LeetCode284 窥探迭代器

《LeetCode之每日一题》:169.顶端迭代器

LeetCode刷题模版:273 - 275278 - 279283 - 284287289 - 290

LeetCode刷题模版:273 - 275278 - 279283 - 284287289 - 290

顶端迭代器