284. 顶端迭代器

---

题目描述

难度：中等

请你设计一个迭代器，除了支持 hasNext 和 next 操作外，还支持 peek 操作。

实现 PeekingIterator 类：

PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素，并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素，返回 true ；否则，返回 false 。
int peek() 返回数组中的下一个元素，但 不 移动指针。

示例：

输入：
["PeekingIterator", "next", "peek", "next", "next", "hasNext"]
[[[1, 2, 3]], [], [], [], [], []]
输出：
[null, 1, 2, 2, 3, false]

解释：
PeekingIterator peekingIterator = new PeekingIterator([1, 2, 3]); // [1,2,3]
peekingIterator.next(); // 返回 1 ，指针移动到下一个元素 [1,2,3]
peekingIterator.peek(); // 返回 2 ，指针未发生移动 [1,2,3]
peekingIterator.next(); // 返回 2 ，指针移动到下一个元素 [1,2,3]
peekingIterator.next(); // 返回 3 ，指针移动到下一个元素 [1,2,3]
peekingIterator.hasNext(); // 返回 False

提示：

1 <= nums.length <= 1000
1 <= nums[i] <= 1000
对 next 和 peek 的调用均有效
next、hasNext 和 peek 最多调用  1000 次

给定实现类：

class PeekingIterator implements Iterator<Integer> {

    public PeekingIterator(Iterator<Integer> iterator) {
        // initialize any member here.
       
    }

    // Returns the next element in the iteration without advancing the iterator.
    public Integer peek() {

    }

    // hasNext() and next() should behave the same as in the Iterator interface.
    // Override them if needed.
    @Override
    public Integer next() {
 
    }

    @Override
    public boolean hasNext() {

    }
}

题解分析

此题相当于给迭代器加上一个方法，该方法能返回当前指针指向的元素；

仔细思考，其实该方法相较于 next 方法也就是少了一步指针后移；

我们可以将构造器中传入的迭代器的所有元素取出放入一个 list，然后维护一个指针模拟 Iterator 即可；

代码：

class PeekingIterator implements Iterator<Integer> {

    private int size = 0;     // 所有元素个数
    private int current = 0;  // 当前元素下标
    private ArrayList<Integer> values = new ArrayList<>();  // 所有元素

    public PeekingIterator(Iterator<Integer> iterator) {
        // initialize any member here.
        while (iterator.hasNext())  values.add(iterator.next());
        size = values.size();
    }

    // Returns the next element in the iteration without advancing the iterator.
    public Integer peek() {
        return values.get(current);  // 返回当前指针指向的元素
    }

    // hasNext() and next() should behave the same as in the Iterator interface.
    // Override them if needed.
    @Override
    public Integer next() {
        return values.get(current++);  // 返回当前指针元素后 指针后移
    }

    @Override
    public boolean hasNext() {
        return current < size;  // 如果当前指针超出了所有元素大小返回 则代表没有下一元素
    }
}

总结

这题其实不难，只要能分析清楚指针在各个方法调用时的位置，就可以写出来；



岁月悠悠，衰微只及肌肤；热忱抛却，颓废必致灵魂