Возвращаемый список узлов дерево в Java-Parent может иметь несколько дочерних узлов
Я пытаюсь написать java-код для возврата списка узлов в дереве.
Дерево выглядит как 
класс узле
class Node{
String label;
List<Node> children;
}
Я пытаюсь таким образом. Но не в состоянии понять, как написать цикл для перебора.
public List<Node> returnAllNodes(Node node){
List<Node> listOfNodes =
new ArrayList<Node>();
boolean iterationCompleted = false;
if(node==null){
return null;
}
while(!iterationCompleted){
if(node.getChildren()==null){
listOfNodes.add(node);
break;
}
else{
//
}
}
return null;
//return traverseAndReturnAllNodes(node.getChildren().get(0));
}
пожалуйста, помогите.
2 ответов
Если вы уверены, что структура дерева (узел не может иметь более одного родителя), это список узлов в порядке убывания глубины:
public static List<Node> returnAllNodes(Node node){
List<Node> listOfNodes = new ArrayList<Node>();
addAllNodes(node, listOfNodes);
return listOfNodes;
}
private static void addAllNodes(Node node, List<Node> listOfNodes) {
if (node != null) {
listOfNodes.add(node);
List<Node> children = node.getChildren();
if (children != null) {
for (Node child: children) {
addAllNodes(child, listOfNodes);
}
}
}
}
Если узлы могут иметь несколько родителей, измените первую строку addAllNodes на:
if (node != null && !listOfNodes.contains(node)) {
первый алгоритм широты выглядит так:
public static List<Node> returnAllNodes(Node node){
List<Node> listOfNodes = new ArrayList<Node>();
if (node != null) {
listOfNodes.add(node);
for(int i = 0; i < listOfNodes.size(); ++i) {
Node n = listOfNodes.get(i);
List<Node> children = n.getChildren();
if (children != null) {
for (Node child: children) {
if (!listOfNodes.contains(child)) {
listOfNodes.add(child);
}
}
}
}
}
return listOfNodes;
}