获取递归嵌套结构的所有叶子元素java中的数据

Question

有人能帮我如何在java中得到递归嵌套结构的所有叶子元素数据吗？

public class Item {
    private String id;
    private List<Item> items;
}

例如：

    A
      - AA
        - aa1
      - AB
        - ab1
          - abab1 
      - a1

后果

在结果中，我只需要得到以下元素列表[aa1，abab1，a1]

首先，感谢您的帮助。

Answer 1

这个怎么样：

public class TestTest {

    public static class Item {
        private final String id;
        private final List<Item> items;

        public Stream<Item> getLeaves() {
            if (items == null || items.isEmpty()) {
                return Stream.of(this);
            }
            return items.stream()
                .flatMap(Item::getLeaves);
        }

    }

    public static void main(String[] args) {
        final var item = new Item(
            "A",
            List.of(
                new Item("AA",
                    List.of(new Item("aa1"))
                ),
                new Item("AB",
                    List.of(new Item("ab1",
                        List.of(new Item("abab1")))
                    )
                ),
                new Item("a1")
            )
        );

        final var leaves = item.getLeaves()
            .map(Item::getId)
            .collect(Collectors.toList());

        System.out.println(leaves);
    }
}

注意：为简洁起见，省略了构造函数和getters

Answer 2

尝试将此方法添加到类：

public List<Item> getLeaves() {
    ArrayList<Item> list = new ArrayList<>();
    if (items.size() == 0) {
        list.add(this);
    } else {
        for (Item item : items) {
            list.addAll(item.getLeaves());
        }
    }
    return list;
}

它只是检查自己是否是一片叶子。如果是，它返回自己，如果不是，它返回它孩子的所有叶子。

Answer 3

你可以用递归的方式很容易地做到：

public static List<Item> findLeaves(Item rootItem){
    List<Item> leaves = new ArrayList<>();
    recursivelyCollectLeaves(leaves, rootItem);
    return leaves;
}

private static void recursivelyCollectLeaves(List<Item> leaves, Item actualItem){
    if(actualItem.getItems().isEmpty()){
        //No children, the actual item is a leaf
        leaves.add(actualItem);
    }
    else{
        for(Item child : actualItem.getItems()){
            recursivelyCollectLeaves(leaves, child);
        }
    }
}

PS：为了让这段代码正常工作，请注意我在Item类中添加了getter。