forked from haoel/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 39
/
PeekingIterator.cpp
88 lines (78 loc) · 2.74 KB
/
PeekingIterator.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
// Source : https://leetcode.com/problems/peeking-iterator/
// Author : Hao Chen
// Date : 2015-11-10
/***************************************************************************************
*
* Given an Iterator class interface with methods: next() and hasNext(), design and
* implement a PeekingIterator that support the peek() operation -- it essentially
* peek() at the element that will be returned by the next call to next().
*
* Here is an example. Assume that the iterator is initialized to the beginning of the
* list: [1, 2, 3].
*
* Call next() gets you 1, the first element in the list.
*
* Now you call peek() and it returns 2, the next element. Calling next() after that
* still return 2.
*
* You call next() the final time and it returns 3, the last element. Calling hasNext()
* after that should return false.
*
* Think of "looking ahead". You want to cache the next element.
* Is one variable sufficient? Why or why not?
* Test your design with call order of peek() before next() vs next() before peek().
* For a clean implementation, check out Google's guava library source code.
*
* Follow up: How would you extend your design to be generic and work with all types,
* not just integer?
*
* Credits:Special thanks to @porker2008 for adding this problem and creating all test
* cases.
*
***************************************************************************************/
// 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 {
private:
bool m_hasNext;
int m_next;
void takeNext() {
m_hasNext = Iterator::hasNext();
if (m_hasNext) {
m_next = Iterator::next();
}
}
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.
takeNext();
}
// Returns the next element in the iteration without advancing the iterator.
int peek() {
return m_next;
}
// hasNext() and next() should behave the same as in the Iterator interface.
// Override them if needed.
int next() {
int temp = m_next;
takeNext();
return temp;
}
bool hasNext() const {
return m_hasNext;
}
};