如何遍历N-Ary树

时间:2015-09-26 09:47:38

标签: java algorithm tree nodes traversal

我的树/节点类:

import java.util.ArrayList;
import java.util.List;

public class Node<T> {
   private T data;
   private List<Node<T>> children;
   private Node<T> parent;

   public Node(T data) {
      this.data = data;
      this.children = new ArrayList<Node<T>>();
   }

   public Node(Node<T> node) {
      this.data = (T) node.getData();
      children = new ArrayList<Node<T>>();
   }

   public void addChild(Node<T> child) {
      child.setParent(this);
      children.add(child);
   }

   public T getData() {
      return this.data;
   }

   public void setData(T data) {
      this.data = data;
   }

   public Node<T> getParent() {
      return this.parent;
   }

   public void setParent(Node<T> parent) {
      this.parent = parent;
   }

   public List<Node<T>> getChildren() {
      return this.children;
   }
}

我知道如何遍历二叉树,但是穿越N-Ary似乎更棘手。

我将如何遍历这棵树。我想要一个计数器,同时我遍历树,以编号/计算树中的每个节点。

然后在特定计数时,我可以停止并返回该计数的节点(可能删除该子树或在该位置添加子树)。

1 个答案:

答案 0 :(得分:3)

最简单的方法是实现这样的访问者模式:

for(std::map<int,std::string>::iterator it = mymap.begin(); it!=mymap.end();)
{
    if(it->first==3)
    {
        it = mymap.erase(it);
    }
    else
    {
        ++it;
    }
}

现在您可以像这样使用它:

public interface Visitor<T> {
    // returns true if visiting should be cancelled at this point
    boolean accept(Node<T> node);
}

public class Node<T> {
    ...

   // returns true if visiting was cancelled
   public boolean visit(Visitor<T> visitor) {
       if(visitor.accept(this))
           return true;
       for(Node<T> child : children) {
           if(child.visit(visitor))
               return true;
       }
       return false;
   }
}

或者您的特定任务:

treeRoot.visit(new Visitor<Type>() {
    public boolean accept(Node<Type> node) {
        System.out.println("Visiting node "+node);
        return false;
    }
});