DFS: добавление узлов с определенным значением в стек (java)

Question

У меня есть дерево. Некоторые узлы в этом дереве имеют значение. Я хочу найти узлы, которые имеют значение, и добавить его в стек.

private Stack<Integer> dfs(Node e, Stack stack) { 

    if(e.num != 0){ 
     stack.push(e.num); 
    } 

    for(Node child : e.children){ 
     dfs(child, stack); 
    } 

    return stack; 
} 

Моя проблема заключается в том, что соответствующие элементы e.num, похоже, не добавляются в стек. Вместо этого стек возвращается пустым.

Answer 1

Вы должны отслеживать посещаемые узлы. ваш код получит исключение, если дерево имеет задний край. Более того, Java является передачей по ссылке, поэтому лучше сделать функцию void.