-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPeekingIterator.java
More file actions
46 lines (39 loc) · 988 Bytes
/
PeekingIterator.java
File metadata and controls
46 lines (39 loc) · 988 Bytes
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
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;
/**
* @description:
* @Author: JachinDo
* @Date: 2019/11/21 22:05
*/
public class PeekingIterator implements Iterator<Integer> {
private List<Integer> list;
private int nextIndex = 0;
public PeekingIterator(Iterator<Integer> iterator) {
list = new ArrayList<Integer>();
while (iterator.hasNext()) {
list.add(iterator.next());
}
}
@Override
public boolean hasNext() {
if (nextIndex <= list.size() - 1) {
return true;
}
return false;
}
@Override
public Integer next() {
if (hasNext()) {
return list.get(nextIndex++);
}
throw new NoSuchElementException();
}
public Integer peek() {
if (hasNext()) {
return list.get(nextIndex);
}
throw new NoSuchElementException();
}
}