我遇到迭代器问题。我正在编写一个自定义链表,使用迭代器来遍历列表。
迭代器看起来像这样:
public class NodeIterator implements Iterator<Node> {
private Node current = head;
private Node lastReturned = head;
public boolean hasNext() {
return lastReturned.getLink() != null;
}
public Node next() {
lastReturned = current;
current = current.getLink();
return lastReturned;
}
public void remove() {
removeNode(lastReturned);
lastReturned = null;
}
}
我还处于早期阶段,因此我通过使用此方法填充节点来从控制台测试数据结构。
private static void MethodToPopulateNodes() {
MyObject o = new MyObject();
String[] responses = new String[prompts.length];
scanner = new Scanner(System.in);
boolean done = false;
String s = null;
while (!done) {
int i = 0;
for (String prompt : prompts) {
System.out.println(prompt);
s = scanner.nextLine();
if (s.equalsIgnoreCase("stop")) {
done = true;
break;
} else {
responses[i] = s;
}
i++;
}
if (done) {
break;
}
o = new MyObject(responses);
myNode.add(c);
}
}
当我只有一个Node时尝试使用迭代器时,它什么都不做。没有错误或任何东西。但是,如果我有多个节点,这个foreach可以完美运行。
public static void main(String[] args) {
myNode = new Node();
methodToPopulateLinkedList();
for (Node node : myNode) {
//toString is overridden for my object
System.out.println(node.getData().toString());
}
}
更新:我编辑了迭代器,在第一次迭代时返回hasNext() == true
:
public class NodeIterator implements Iterator<Node> {
private boolean done = false;
private Node current = head;
private Node lastReturned = head;
public boolean hasNext() {
if (head == tail && head != null && !done) {
done = true;
return true;
}
return lastReturned.getLink() != null;
}
public Node next() {
lastReturned = current;
current = current.getLink();
return lastReturned;
}
public void remove() {
removeNode(lastReturned);
lastReturned = null;
}
}
我觉得这是超级的,但它确实有效。似乎Java在调用next之前首先调用hasNext()
所以我必须区别对待特殊情况。
| 123
hasNext() == true
next() == 1
1 | 23
hasNext() == true
next() == 2
12 | 3
|
等于光标的位置。那是准确的吗?有没有更好的方法来解决这个问题?
答案 0 :(得分:2)
如果只有一个节点,那么它的特殊情况是 - &gt;接下来是空的。在循环之前,尝试打印出第一个节点,我认为你的循环可能正在向前看。