我是如何给阿里大神Tree工具类做CodeReview并优化的

最新推荐文章于 2024-10-09 08:32:56 发布

赵侠客

最新推荐文章于 2024-10-09 08:32:56 发布

阅读量885

点赞数 13

CC 4.0 BY-SA版权

分类专栏： JAVA常用代码块文章标签： java 代码复审数据结构

本文链接：https://blog.csdn.net/whzhaochao/article/details/140858428

首发公众号：赵侠客

引言

前段时间我写了一篇关于树操作的工具类《解密阿里大神写的天书般的Tree工具类，轻松搞定树结构！》，当时主要把精力集中在分析代码的实现层面，没有从设计层面、性能层考虑，然后就被很多网友大神喷了，在此我做下深刻的反思，同时也针对大神网友提的缺点做出优化。
主要的缺点：

性能问题，因为使用递归，最坏的情况时间复杂度为 O(n^n)，平均为O(nlogn)，性能确实非常差
使用了peek()，peak()坑确实比较多，官方建议仅在调试和日志记录时使用，避免在 peek() 中修改元素状态
没有filter、search等方法，过滤和搜索也是树形结构中常用的方法

本文接针对这三点做了优化

一、性能优化

1.1 时间复杂度降到O(n)

使用网友的建议合成树使用Map用空间来换时间，将时间复杂度降到O(n)，我们直接看优化后的代码：

public static <T, E> List<E> makeTree(List<E> menuList, Function<E, T> pId, Function<E, T> id, Predicate<E> rootCheck, BiConsumer<E, List<E>> setSubChildren) {
   
   
    //按原数组顺序构建父级数据Map，使用Optional考虑pId为null
    Map<Optional<T>, List<E>> parentMenuMap = menuList.stream().collect(Collectors.groupingBy(
            node -> Optional.ofNullable(pId.apply(node)),
            LinkedHashMap::new,
            Collectors.toList()
    ));
    List<E> result = new ArrayList<>();
    for (E node : menuList) {
   
   
        //添加到下级数据中
        setSubChildren.accept(node, parentMenuMap.get(Optional.ofNullable(id.apply(node))));
        //如里是根节点，加入结构
        if (rootCheck.test(node)) {
   
   
            result.add(node);
        }
    }
    return result;
}

方法参数说明：

List menuList,需要合成的集全数据
Function<E, T> pId，实体中的父级ID字段，可以为null，如：MenuVo::getPId
Function<E, T> id，实体中的ID字段，不能为null，如，MenuVo::getId
Predicate rootCheck，判断为根节点条件，如： x->x.getPId()==-1L
BiConsumer<E, List> setSubChildren，设置子节点方法，如：MenuVo::setSubMenus

使用方法：

//定义GroupVo
public class GroupVo {
   
   
    private String groupId;
    private String parentGroupId;
    private List<GroupVo> subGroups;
}
//测试合成树
GroupVo groupVo1=new GroupVo("a",null);
GroupVo groupVo2=new GroupVo("b",null);
GroupVo groupVo3=new GroupVo("c","a");
GroupVo groupVo4=new GroupVo("d","b");
List<GroupVo> groupVos= Arrays.asList(groupVo1,groupVo2,groupVo3,groupVo4);
List<GroupVo> tree=TreeUtil.makeTree(groupVos,GroupVo::getParentGroupId,GroupVo::getGroupId,x->x.getParentGroupId()==null,GroupVo::setSubGroups);
System.out.println(JsonUtils.toJson(tree));

输出结果：

[
    {
   
   
        "id": "a",
        "subGroup": [
            {
   
   
                "id": "c",
                "pid": "a"
            }
        ]
    },
    {
   
   
        "id":