当前位置: 代码迷 >> 综合 >> 1005 lc.284 iterator设计
  详细解决方案

1005 lc.284 iterator设计

热度:59   发布时间:2023-11-22 10:44:01.0

lc.284 顶端迭代器

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

实现 PeekingIterator 类:

PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素,并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素,返回 true ;否则,返回 false 。
int peek() 返回数组中的下一个元素,但 不 移动指针。
  • C++
/** Below is the interface for Iterator, which is already defined for you.* **DO NOT** modify the interface for Iterator.** class Iterator {* struct Data;* Data* data;* public:* Iterator(const vector<int>& nums);* Iterator(const Iterator& iter);** // Returns the next element in the iteration.* int next();** // Returns true if the iteration has more elements.* bool hasNext() const;* };*/class PeekingIterator : public Iterator {
    
public:int nextElement;bool flag;PeekingIterator(const vector<int>& nums) : Iterator(nums) {
    flag = Iterator::hasNext();if (flag) {
    nextElement = Iterator::next();}}int peek() {
    return nextElement;}int next() {
    int ret = nextElement;flag = Iterator::hasNext();if (flag) {
    nextElement = Iterator::next();}return ret;}bool hasNext() const {
    return flag;}};

C++ 中 PeekingIterator 继承父类 Iterator,Iterator 已经实现方法 next 和 hasNext,在此我们在 PeekingIterator 中主要实现 peek方法即可。我们使用 flag 标记迭代器是否还有剩余元素,使用 nextElement 存储迭代器的下一个元素。

next:首先用 ret存储 nextElement表示返回值,flag 保存 Iterator 调用 hasNext方法的返回结果,然后将 nextElement 向后移动一位,最后返回 ret;hasNext:返回 flag;peek:由于 peek\操作不改变指针,因此返回 nextElement。
  相关解决方案