Как уже упоминалось, создайте хэш-таблицу / карту объектов для списка их (прямых) дочерних элементов.
Отсюда вы можете легко найти список прямых потомков вашего «целевого объекта», а затем повторить процесс для каждого объекта в списке.
Вот как я это сделал на Java и с использованием обобщений, с очередью вместо любой рекурсии:
public static Set<Node> findDescendants(List<Node> allNodes, Node thisNode) {
// keep a map of Nodes to a List of that Node's direct children
Map<Node, List<Node>> map = new HashMap<Node, List<Node>>();
// populate the map - this is O(n) since we examine each and every node
// in the list
for (Node n : allNodes) {
Node parent = n.getParent();
if (parent != null) {
List<Node> children = map.get(parent);
if (children == null) {
// instantiate list
children = new ArrayList<Node>();
map.put(parent, children);
}
children.add(n);
}
}
// now, create a collection of thisNode's children (of all levels)
Set<Node> allChildren = new HashSet<Node>();
// keep a "queue" of nodes to look at
List<Node> nodesToExamine = new ArrayList<Node>();
nodesToExamine.add(thisNode);
while (nodesToExamine.isEmpty() == false) {
// pop a node off the queue
Node node = nodesToExamine.remove(0);
List<Node> children = map.get(node);
if (children != null) {
for (Node c : children) {
allChildren.add(c);
nodesToExamine.add(c);
}
}
}
return allChildren;
}
Ожидаемое время выполнения - что-то между O (n) и O (2n), если я помню, как рассчитать это право. Вы гарантированно просматриваете каждый узел в списке, плюс еще несколько операций, чтобы найти всех потомков вашего узла - в худшем случае (если вы запускаете алгоритм на корневом узле) вы смотрите на каждый узел список дважды.