mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2026-08-30 19:44:57 +08:00
add 284 🍺
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
# [284. Peeking Iterator](https://leetcode.com/problems/peeking-iterator/)
|
||||
|
||||
# 思路
|
||||
|
||||
我们有了一个只支持`next`和`hasNext`方法的迭代器类,让我们实现一个子类,要求增添一个`peek`方法。
|
||||
|
||||
## 思路一
|
||||
我们想实现的`peek`是只返回当前元素而不考虑指针后移。但每次调用基类的`next`时,返回当前元素后都会后移指针。那么我们可以考虑用一个变量`value`在调用基类的`next`时缓存当前元素,这样的话即使指针后移了,我们调用`peek`时也能直接返回value。为此我们需要一个bool变量`flag`来指示`value`是否存放了在之后调用`peek`和`next`应该返回的值。
|
||||
|
||||
|
||||
## 思路二
|
||||
还有一个比较tricky的方法:`peek`时创建一个当前指针对应的副本,由于是局部变量,所以调用结束后会被销毁,所以并没有任何内存问题。
|
||||
|
||||
# C++
|
||||
|
||||
## 思路一
|
||||
``` C++
|
||||
class PeekingIterator : public Iterator {
|
||||
private:
|
||||
bool flag;
|
||||
int value;
|
||||
|
||||
public:
|
||||
PeekingIterator(const vector<int>& nums) : Iterator(nums) {
|
||||
flag = false; // value是否存放了peek和next应该返回的值
|
||||
}
|
||||
|
||||
int peek(){
|
||||
if(!flag) value = Iterator::next();
|
||||
flag = true;
|
||||
return value;
|
||||
}
|
||||
|
||||
int next() {
|
||||
if(!flag) return Iterator::next();
|
||||
flag = false;
|
||||
return value;
|
||||
}
|
||||
|
||||
bool hasNext() const {
|
||||
return flag || Iterator::hasNext();
|
||||
}
|
||||
};
|
||||
```
|
||||
|
||||
## 思路二
|
||||
``` C++
|
||||
class PeekingIterator : public Iterator {
|
||||
public:
|
||||
PeekingIterator(const vector<int>& nums) : Iterator(nums) {}
|
||||
|
||||
int peek(){
|
||||
// 调用Iterator(const Iterator& iter)创造一个副本
|
||||
return Iterator(*this).next();
|
||||
}
|
||||
|
||||
// int next() {}
|
||||
// bool hasNext() const {}
|
||||
};
|
||||
```
|
||||
|
||||
Reference in New Issue
Block a user