使用函数式将List集合转化Tree

最新推荐文章于 2024-07-13 15:17:36 发布

好雨知时节！

最新推荐文章于 2024-07-13 15:17:36 发布

阅读量358

点赞数 10

文章标签： list java

本文链接：https://blog.csdn.net/qq_35325817/article/details/136233631

版权

先创建TreeBuilder实例的静态方法

@FunctionalInterface
interface TreeBuilder {
    // 构建树的方法
    List<ObjectConverter.Node> buildTree(List<ObjectConverter.Node> nodes, Integer parentID);

    // 创建TreeBuilder实例的静态方法
    static TreeBuilder createTree() {
        return (nodes, parentID) -> {
            // 将节点按照父节点ID分组
            Map<Integer, List<ObjectConverter.Node>> map = groupByKey(nodes, ObjectConverter.Node::getParentId);

            // 将每个节点的子节点列表赋值
            nodes.forEach(node -> node.children = map.getOrDefault(node.id, new ArrayList<>()));

            // 返回指定父节点ID的节点列表
            return nodes.stream().filter(node -> node.parentId != null && node.parentId.equals(parentID)).collect(Collectors.toList());
        };
    }

    // 根据指定的键函数将列表分组
    static <E, K> Map<K, List<E>> groupByKey(List<E> list, Function<E, K> function) {
        return CollUtil.isEmpty(list) ? new HashMap<>() : list.stream().collect(Collectors.groupingBy(function, HashMap::new, Collectors.toList()));
    }
}

调用方式伪代码

public class ObjectConverter {

    @Data
    @NoArgsConstructor
    static class Node {
        int id;
        Integer parentId;
        String name;
        List<Node> children;

        public Node(int id, Integer parentId, String name) {
            this.id = id;
            this.parentId = parentId;
            this.name = name;
            this.children = new ArrayList<>();
        }

        public Function<Node,Integer> getParentId1() {
            return Node::getParentId;
        }
    }


    public static void main(String[] args) {
        Node node = new Node();
        Function<Node, Integer> parentId = node.getParentId1();

        // 假设这是你的初始列表
        List<ObjectConverter.Node> list = new ArrayList<>();
        list.add(new ObjectConverter.Node(1, 0, "Node 1"));
        list.add(new ObjectConverter.Node(2, 1, "Node 1.1"));
        list.add(new ObjectConverter.Node(3, 2, "Node 1.2"));
        list.add(new ObjectConverter.Node(4, 2, "Node 1.1.1"));
        list.add(new ObjectConverter.Node(5, 3, "Node 1.2.1"));
        // 使用自定义的函数式接口来构建树形结构
        TreeBuilder treeBuilder = TreeBuilder.createTree();
        List<ObjectConverter.Node> tree = treeBuilder.buildTree(list, 0);
        // 输出转换后的树形结构
        System.out.println(JSONObject.toJSONString(tree));
    }

}