Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
peeking-iterator.cpp 1.38 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2015-09-21 09:56 +08:00 . Update peeking-iterator.cpp
// Time: O(1) per peek(), next(), hasNext()
// Space: O(1)
// 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), has_next_(Iterator::hasNext()) {
// Initialize any member here.
// **DO NOT** save a copy of nums and manipulate it directly.
// You should only use the Iterator interface methods.
}
// Returns the next element in the iteration without advancing the iterator.
int peek() {
if (!has_peeked_) {
has_peeked_ = true;
val_ = Iterator::next();
}
return val_;
}
// hasNext() and next() should behave the same as in the Iterator interface.
// Override them if needed.
int next() {
val_ = peek();
has_peeked_ = false;
has_next_ = Iterator::hasNext();
return val_;
}
bool hasNext() const {
return has_next_;
}
private:
int val_;
bool has_next_;
bool has_peeked_ = false;
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助