生成树结构

开发中会遇到很多需要层级展示的功能,这就需要一个树形结构数据来配合前端的渲染,网上查了很多,大都是递归方式实现的,下面是非递归方式的实现:

返回数据的结构:

@Data
@Accessors(chain = true)
public class CourseTypeTreeVo {

    /**
     * id
     */
    private Long id;

    /**
     * 父id
     */
    private Long parentId = 0L;

    /**
     * 层级
     */
    private Integer level = 1;

    /**
     * 分类名
     */
    private String name;

    /**
     * 子分类
     */
    private List<CourseTypeTreeVo> childrenCourseType;

}

传入的数据结构:

@Data
@Accessors(chain = true)
public class CourseType implements Serializable {

    private static final long serialVersionUID = 5596324500728175163L;

    /**
     * 分类id
     */
    private Long id;

    /**
     * 分类父id,默认0表示没有父级
     */
    private Long parentId = 0L;

    /**
     * 租户id
     */
    private Long tenantId;

    /**
     * 层级
     */
    private Integer level;

    /**
     * 分类名称
     */
    private String name;

    /**
     * 创建人
     */
    @TableField(fill = FieldFill.INSERT)
    private Long createdBy;

    /**
     * 创建时间
     */
    @TableField(fill = FieldFill.INSERT)
    private LocalDateTime createdTime;

    /**
     * 更新人
     */
    @TableField(fill = FieldFill.UPDATE)
    private Long updatedBy;

    /**
     * 更新时间
     */
    @TableField(fill = FieldFill.UPDATE)
    private LocalDateTime updatedTime;

    /**
     * 删除标识 1=未删除 2 已删除
     */
    private Integer delFlag;

}

一顿操作成树:

private List<CourseTypeTreeVo> toTree(List<CourseType> courseTypes) {
        // 对象类型转换
        List<CourseTypeTreeVo> list = new ArrayList<>();
        for (CourseType item : courseTypes) {
            CourseTypeTreeVo courseTypeTreeVo = new CourseTypeTreeVo();
            // 用的hutool的
            BeanUtil.copyProperties(item, courseTypeTreeVo);
            list.add(courseTypeTreeVo);
        }

        // 根节点
        List<CourseTypeTreeVo> result = list.stream().filter(item -> 1 == item.getLevel()).collect(Collectors.toList());

        // id->实体
        Map<Long, CourseTypeTreeVo> mapById = list.stream().collect(Collectors.toMap(CourseTypeTreeVo::getId, CourseTypeTreeVo -> CourseTypeTreeVo));
        // children
        list.removeAll(result);
        // 按parentId分组
        Map<Long, List<CourseTypeTreeVo>> mapByParentId = list.stream().collect(Collectors.groupingBy(CourseTypeTreeVo::getParentId));
        // 关系绑定
        for (Long key : mapByParentId.keySet()) {
            mapById.get(key).setChildrenCourseType(mapByParentId.get(key));
        }

        return result;
    }

基本思路是按父节点分组,把同属于一个父节点的集合赋值给父节点,绑定关系的时间复杂度最好是O(1),最坏是O(n-1),有没有更好的方式把这块再优化一下

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值