【LeetCode】284. 顶端迭代器结题报告 (C++)

最新推荐文章于 2024-04-15 11:36:17 发布

暮雨凉初透

最新推荐文章于 2024-04-15 11:36:17 发布

阅读量412

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/qq_32805671/article/details/84888070

LeetCode 专栏收录该内容

132 篇文章

订阅专栏

本文解析了LeetCode上的一道迭代器设计题目，详细介绍了如何将原有的迭代器类扩展，增加peek()方法，使其能在不改变迭代状态的情况下预览下一个元素。通过具体示例和代码实现，展示了如何在迭代器中实现peek()、next()和hasNext()方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题地址：https://leetcode-cn.com/problems/peeking-iterator/

题目描述：

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

示例:

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

调用 next() 返回 1，得到列表中的第一个元素。
现在调用 peek() 返回 2，下一个元素。在此之后调用 next() 仍然返回 2。
最后一次调用 next() 返回 3，末尾元素。在此之后调用 hasNext() 应该返回 false。

解题方案：

设计题型，写起来很简单。对迭代器进行改造。

代码：

// 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);
	virtual ~Iterator();
	// 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:
	PeekingIterator(const vector<int>& nums) : Iterator(nums) {
	    // Initialize any member here.
	    // **DO NOT** save a copy of nums and manipulate it directly.
	    // You should only use the Iterator interface methods.
	    _flag = false;
	}
 
    // Returns the next element in the iteration without advancing the iterator.
	int peek() {
        if (!_flag) {
            _value = Iterator::next();
            _flag = true;
        }
        return _value;
	}
 
	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	int next() {
	    if (!_flag) return Iterator::next();
	    _flag = false;
	    return _value;
	}
 
	bool hasNext() const {
	    if (_flag) return true;
	    if (Iterator::hasNext()) return true;
	    return false;
	}
private:
    int _value;
    int _flag;
};