diff options
Diffstat (limited to 'other/peeking_iterator.cpp')
-rw-r--r-- | other/peeking_iterator.cpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/other/peeking_iterator.cpp b/other/peeking_iterator.cpp new file mode 100644 index 0000000..1c4d0b9 --- /dev/null +++ b/other/peeking_iterator.cpp @@ -0,0 +1,32 @@ +#include "leetcode.h" + +class Iterator { + struct Data; + Data *data; + +public: + Iterator(const vector<int> &nums); + Iterator(const Iterator &iter); + int next(); + bool hasNext() const; +}; + +class PeekingIterator : public Iterator { +public: + int nextVal; + PeekingIterator(const vector<int> &nums) : Iterator(nums) { + nextVal = Iterator::next(); + } + int peek() { return nextVal; } + int next() { + int tmp = nextVal; + if (Iterator::hasNext()) + nextVal = Iterator::next(); + else + nextVal = NULL; + return tmp; + } + bool hasNext() const { return (nextVal != NULL); } +}; + +int main() {} |