为什么我的二叉树迭代器会停止?

时间:2014-04-09 12:09:56

标签: java tree

在尝试以顺序方式遍历二叉树时,它会打印根,最右边的子项然后停止。有些事情显然是错的,而且我很难绕过这个问题,因为我是Tree-structures的新手。我做错了什么?

主要

while (tree.iterator().hasNext())
    System.out.println(tree.iterator().next());

迭代

public Iterator<T> iterator() {
  Iterator<T> it = new Iterator<T>() {

    Node<T> next = root;

    @Override
    public boolean hasNext() {
      return next != null;
    }

    @Override
    public T next() {
      if (!hasNext())
        throw new NoSuchElementException();

      System.out.println("Returning data");
      T r = next.data;

      if (next.right != null) {
        next = next.right;
        while (next.left != null)
          next = next.left;

      } else while (true) {
        if (next.parent == null) {
          next = null;
          return r;
        }
        if (next.parent.left == next) {
          next = next.parent;
          return r;
        }
        next = next.parent;
      }
      return r;
    }

    @Override
    public void remove() {
      // TODO Auto-generated method stub
    }

  };
  return it;
}

输出:

242
275
279
283

242是树的根。
242.left = 33
242.right = 283

更新1:


242
|33
||29
|||25
||||NULL
||||NULL
|||NULL
||74
|||70
||||66
|||||NULL
|||||NULL
||||NULL
|||115
||||111
|||||107
||||||NULL
||||||NULL
|||||NULL
||||156
|||||152
||||||148
|||||||NULL
|||||||NULL
||||||NULL
|||||197
||||||193
|||||||NULL
|||||||NULL
||||||238
|||||||234
||||||||NULL
||||||||NULL
|||||||NULL
|283
||279
|||275
||||NULL
||||NULL
|||NULL
||NULL

1 个答案:

答案 0 :(得分:2)

在前往最左边的孩子之前,你似乎从根的右侧开始。所以你错过了整个左侧部分。

您应该使用最左边的孩子初始化next,而不是根。

更新:您可以在初始化块中执行此操作,如下所示:

Iterator<T> it = new Iterator<T>() {

    Node<T> next;

    {
        next = root;
        while (next.left != null)
            next = next.left;
    }

    @Override
    public boolean hasNext() {
        return next != null;     
    }

    //...
}