Я пишу программу, которая берет двоичное дерево и перебирает его по порядку. Для этого мой класс-итератор проходит по дереву и добавляет элементы в список, после чего список повторяется. При отладке своего кода я обнаружил, что список содержит 4 отдельных элемента, как и должно быть, но в main
, когда список печатается, я снова и снова получаю tinsel
(первый элемент списка) в бесконечном цикле. Следовательно, итератор просто не выполняет итерацию; он застревает на первом элементе.
Класс итератора:
import java.util.*;
@SuppressWarnings("rawtypes")
public class TreeIterator implements Iterator{
BinaryTree BT;
TreeIterator(BinaryTree BT){
this.BT=BT;
inOrder(BT.root);
}
private List<String> decorationList = new ArrayList<String>();
private void inOrder(Node root){
if(root==null) return;
inOrder(root.left);
String temp = root.decoration;
decorationList.add(temp);
inOrder(root.right);
}
public boolean hasNext() {
return decorationList.iterator().hasNext();
}
public Object next() {
return decorationList.iterator().next();
}
public void remove() {
// this method is not implemented
}
}
Основная функция:
public class Main {
public static void main(String[] args) {
// build a test tree
//
// star
// / \
// tinsel red balls
// \
// lights
BinaryTree BT = new BinaryTree();
BT.root = new Node();
BT.root.decoration = "star";
BT.root.left = new Node();
BT.root.left.decoration = "tinsel";
BT.root.left.right = new Node();
BT.root.left.right.decoration = "lights";
BT.root.right = new Node();
BT.root.right.decoration = "red balls";
TreeIterator TI = BT.createIterator();
while(TI.hasNext()){
System.out.println(TI.next());
}
}
}
Дайте мне знать, следует ли мне также добавить реализацию двоичного дерева.
@SuppressWarnings("rawtypes")
зло. Кроме того, почему вы снова получаетеiterator
при каждом звонке? - person Elliott Frisch   schedule 20.11.2014