forked from DengWangBao/Leetcode-Java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathVector2D.java
More file actions
32 lines (27 loc) · 782 Bytes
/
Vector2D.java
File metadata and controls
32 lines (27 loc) · 782 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
import java.util.Iterator;
import java.util.List;
public class Vector2D implements Iterator<Integer> {
private Iterator<List<Integer>> mIterator;
private Iterator<Integer> mSubIterator;
public Vector2D(List<List<Integer>> vec2d) {
mIterator = vec2d.iterator();
}
@Override
public Integer next() {
if (hasNext()) {
return mSubIterator.next();
}
return 0;
}
@Override
public boolean hasNext() {
while (mSubIterator == null || !mSubIterator.hasNext()) {
if (mIterator.hasNext()) {
mSubIterator = mIterator.next().iterator();
} else {
break;
}
}
return mSubIterator != null && mSubIterator.hasNext();
}
}