Java怎么将Excel数据转化为树形结构

在日常的开发工作中，我们经常会遇到需要将Excel表格中的数据转化为树形结构的需求。这种需求在数据展示、数据分析和数据存储等场景中非常常见。本文将详细介绍如何使用Java将Excel数据转化为树形结构。

1. 准备工作

在开始之前，我们需要准备以下工具和库：

• Apache POI：用于读取和写入Excel文件。
• Java集合框架：用于存储和操作树形结构。

1.1 添加依赖

如果你使用的是Maven项目，可以在pom.xml中添加以下依赖：

<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi</artifactId>
    <version>5.2.3</version>
</dependency>
<dependency>
    <groupId>org.apache.poi</groupId>
    <artifactId>poi-ooxml</artifactId>
    <version>5.2.3</version>
</dependency>

2. 读取Excel数据

首先，我们需要使用Apache POI读取Excel文件中的数据。假设我们有一个Excel文件，其中包含以下数据：

2.1 读取Excel文件

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class ExcelReader {
    public static List<Node> readExcel(String filePath) throws IOException {
        List<Node> nodes = new ArrayList<>();
        FileInputStream file = new FileInputStream(filePath);
        Workbook workbook = new XSSFWorkbook(file);
        Sheet sheet = workbook.getSheetAt(0);

        for (Row row : sheet) {
            if (row.getRowNum() == 0) continue; // 跳过表头
            int id = (int) row.getCell(0).getNumericCellValue();
            String name = row.getCell(1).getStringCellValue();
            int parentId = (int) row.getCell(2).getNumericCellValue();
            nodes.add(new Node(id, name, parentId));
        }

        workbook.close();
        file.close();
        return nodes;
    }
}

2.2 定义Node类

public class Node {
    private int id;
    private String name;
    private int parentId;
    private List<Node> children;

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

    // Getters and Setters
}

3. 构建树形结构

接下来，我们需要将读取到的数据转化为树形结构。我们可以通过递归的方式来实现这一点。

3.1 构建树形结构

import java.util.List;

public class TreeBuilder {
    public static Node buildTree(List<Node> nodes) {
        Node root = null;
        for (Node node : nodes) {
            if (node.getParentId() == 0) {
                root = node;
            } else {
                Node parent = findParent(root, node.getParentId());
                if (parent != null) {
                    parent.getChildren().add(node);
                }
            }
        }
        return root;
    }

    private static Node findParent(Node root, int parentId) {
        if (root.getId() == parentId) {
            return root;
        }
        for (Node child : root.getChildren()) {
            Node result = findParent(child, parentId);
            if (result != null) {
                return result;
            }
        }
        return null;
    }
}

3.2 测试代码

public class Main {
    public static void main(String[] args) throws IOException {
        List<Node> nodes = ExcelReader.readExcel("data.xlsx");
        Node root = TreeBuilder.buildTree(nodes);
        printTree(root, 0);
    }

    private static void printTree(Node node, int level) {
        for (int i = 0; i < level; i++) {
            System.out.print("--");
        }
        System.out.println(node.getName());
        for (Node child : node.getChildren()) {
            printTree(child, level + 1);
        }
    }
}

4. 总结

通过以上步骤，我们成功地将Excel中的数据转化为树形结构。首先，我们使用Apache POI读取Excel文件中的数据，然后通过递归的方式构建树形结构。最后，我们可以通过遍历树形结构来展示数据。

这种方法不仅适用于简单的树形结构，还可以扩展到更复杂的场景中。希望本文对你有所帮助！