java list 按多个字段分组转tree

原创

mob64ca12f10f72 2024-01-14 06:54:45 ©著作权

文章标签 List 字段 Java 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f10f72的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java List 按多个字段分组转Tree

在Java开发中，我们经常会遇到需要对一个List进行分组的情况。而有时，我们不仅需要按照一个字段进行分组，还需要按照多个字段进行分组。本文将介绍如何使用Java实现对List按多个字段分组，并将其转化为树形结构。

什么是List？

在Java中，List是一种常见的数据结构，可以存储多个元素，并且可以按照插入的顺序进行访问。List是一种有序集合，可以包含重复的元素。

什么是分组？

在数据处理中，分组是一种将数据按照某些特定的标准进行分类的方法。例如，我们有一个包含学生信息的List，可以按照年级、班级等字段进行分组，将相同年级或班级的学生归为一组。

实现思路

要实现对List按多个字段进行分组，并将其转化为树形结构，我们可以使用Java中的Map和递归来实现。

首先，我们需要定义一个树形结构的节点类，该节点类包含一个存储数据的字段和一个子节点列表。
然后，我们遍历List中的每个元素，根据多个字段的值创建对应的节点，并将该节点添加到对应的父节点的子节点列表中。
最后，我们将根节点添加到一个Map中，以便于后续的操作。

示例代码

下面是一个示例代码，演示了如何将一个包含学生信息的List按照年级和班级进行分组，并将其转化为树形结构。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class Student {
    private String name;
    private String grade;
    private String className;

    public Student(String name, String grade, String className) {
        this.name = name;
        this.grade = grade;
        this.className = className;
    }

    // Getters and setters
}

class TreeNode {
    private String key;
    private List<TreeNode> children;

    public TreeNode(String key) {
        this.key = key;
        children = new ArrayList<>();
    }

    // Getters and setters
}

public class GroupByMultipleFields {
    public static void main(String[] args) {
        List<Student> students = new ArrayList<>();
        students.add(new Student("Alice", "A", "Class1"));
        students.add(new Student("Bob", "A", "Class1"));
        students.add(new Student("Charlie", "B", "Class2"));
        students.add(new Student("David", "B", "Class2"));

        Map<String, TreeNode> resultMap = new HashMap<>();

        for (Student student : students) {
            String key = student.getGrade() + "-" + student.getClassName();
            TreeNode node = resultMap.getOrDefault(key, new TreeNode(key));
            node.getChildren().add(new TreeNode(student.getName()));
            resultMap.put(key, node);
        }

        // 输出结果
        for (TreeNode node : resultMap.values()) {
            System.out.println(node.getKey());
            for (TreeNode child : node.getChildren()) {
                System.out.println("  " + child.getKey());
            }
        }
    }
}

输出结果为：

A-Class1
  Alice
  Bob
B-Class2
  Charlie
  David

状态图

下面是一个状态图，展示了如何将List按多个字段进行分组，并将其转化为树形结构。

stateDiagram
    [*] --> 初始状态
    初始状态 --> 创建根节点
    创建根节点 --> 遍历列表
    遍历列表 --> 创建节点
    创建节点 --> 添加到父节点
    添加到父节点 --> 遍历列表
    遍历列表 --> 输出结果
    输出结果 --> [*]