Как получить все листья данного узла? (дерево имеет только id_parent)

Question

У меня есть ArrayList<Foo> структура в Java со стандартными древовидными полями: id, id_parent, value

Я хочу получить список всех листьев (id_elements) данного узла.

Мой пример данных:

Foo ArrayList<E>
    elementData Object[10]
        [0] Foo
            id          1333393146
            id_element  1333398441937
            id_parent   1333393120
            value       "1.1."
        [1] Foo
            id          1333393120
            id_element  0
            id_parent   0
            value       "1."
        [2] Foo
            id          1333400050
            id_element  0
            id_parent   0
            value       "2."
        [3] Foo
            id          1333400480
            id_element  0
            id_parent   1333400050
            value       "2.1."
        [4] Foo
            id          1333400596
            id_element  1335957085269
            id_parent   1333400480
            value       "2.1.1."
        [5] Foo
            id          1333401059
            id_element  1335957088564
            id_parent   1333400480
            value       "2.1.2."
        [6] Foo
            id          1333401973
            id_element  1335957090492
            id_parent   1333400480
            value       "2.1.3."
        [7] Foo
            id          1333401974
            id_element  1335957093220
            id_parent   1333400050
            value       "2.2."
        [8] Foo
            id          1333392031
            id_element  0
            id_parent   0
            value       "3."
        [9] Foo
            id          1333394672
            id_element  1335957098326
            id_parent   1333392031
            value       "3.1."

Мне нужно сделать функцию public ArrayList<Long> GetIds(ArrayList<Foo> tree, Long id_node) { } где tree это моя структура, и id_node это идентификатор узла.

Мне нужно только родительский узел оставляет не дочерние узлы.

например:

---

вход: [вышеуказанная структура], id = 1333400050

выход: 1335957085269, 1335957088564, 1335957090492, 1335957093220

---

Я не знаю, почему у меня есть затмение по этому поводу..

Answer 1

public static ArrayList<Long> getIds(ArrayList<Foo> tree, Long id_node) {
    ArrayList<Long> leaves = new ArrayList<Long>();
    for (Foo foo : tree) {
        if (foo.id_parent == id_node) {
            ArrayList<Long> ids = getIds(tree, foo.id);
            if (ids == null) {
                leaves.add(foo.id);
            } else {
                leaves.addAll(ids);
            }
        }
    }
    if (leaves.isEmpty()) {
        return null;
    }
    return leaves;
}