如何将List转换为有序的Tree结构？

高效实现列表到有序树的转换

本文探讨如何将列表数据转换为有序的树状结构。 原始方法忽略了排序需求，本文将改进算法，使其能够根据指定属性对树节点进行排序。

排序策略

我们采用Comparator接口进行排序。通过自定义比较器，根据节点的特定属性（例如sort字段）进行排序。在列表转换为树之前，先对列表进行排序，确保树结构的有序性。

改进后的代码如下：

// 列表转换为树结构
private List listToTree(List list) {
    // 根据 sort 属性升序排序
    list.sort(Comparator.comparing(MenuNode::getSort));

    List tree = new ArrayList<>();
    for (MenuNode menu : list) {
        // 查找根节点
        if (menu.getParentCode().isEmpty()) {
            tree.add(menu);
        }

        // 查找子节点
        List children = new ArrayList<>();
   
     for (MenuNode node : list) {
            if (node.getParentCode().equals(menu.getMenuCode())) {
                children.add(node);
            }
        }
        menu.setChildren(children);
    }

    return tree;
}

为了避免同级节点排序冲突，请确保sort属性在数据源中具有唯一值。 此方法确保了树结构在构建过程中就已完成排序，提高了效率和可读性。