Java JSONArray类型数据转树结构

在Java开发中，我们常常需要将一个JSONArray类型的数据转换为树结构。这在某些业务场景下非常有用，比如处理菜单数据、权限管理等。本文将介绍如何使用Java进行JSONArray类型数据的转树结构，并提供代码示例供读者参考。

什么是JSONArray？

在介绍JSONArray类型数据的转树结构之前，我们先来了解下什么是JSONArray。JSONArray是Java中的一个类，位于org.json包中，用于表示一个JSON数组。它可以存储多个JSON对象，通过索引访问每个JSON对象。

树结构的基本概念

在计算机科学中，树是一种非常常见的数据结构。它由节点（node）构成，每个节点可以有零个或多个子节点，节点之间存在一对多的关系。树的顶部节点称为根节点（root），没有子节点的节点称为叶子节点（leaf），非根节点也非叶子节点的节点称为内部节点（internal node）。

JSONArray数据转树结构的思路

要将JSONArray类型的数据转换为树结构，我们可以采用以下的思路：

  1. 遍历JSONArray中的每个JSON对象。
  2. 根据JSON对象中的父节点ID（或其他标识），将节点加入对应的父节点。
  3. 如果节点没有父节点，那么它就是根节点。

代码示例

下面是一个简单的代码示例，演示了如何将JSONArray类型的数据转换为树结构：

import org.json.JSONArray;
import org.json.JSONObject;

import java.util.HashMap;
import java.util.Map;

public class JSONArrayToTree {

    public static void main(String[] args) {
        // 假设我们有以下的JSONArray数据
        JSONArray jsonArray = new JSONArray("[{\"id\":1,\"name\":\"Node 1\",\"parentId\":0},{\"id\":2,\"name\":\"Node 2\",\"parentId\":1},{\"id\":3,\"name\":\"Node 3\",\"parentId\":1},{\"id\":4,\"name\":\"Node 4\",\"parentId\":0},{\"id\":5,\"name\":\"Node 5\",\"parentId\":4}]");

        Map<Integer, TreeNode> nodeMap = new HashMap<>();

        // 构建树结构
        for (int i = 0; i < jsonArray.length(); i++) {
            JSONObject jsonObject = jsonArray.getJSONObject(i);
            int id = jsonObject.getInt("id");
            String name = jsonObject.getString("name");
            int parentId = jsonObject.getInt("parentId");

            TreeNode node = new TreeNode(id, name);
            nodeMap.put(id, node);

            if (parentId != 0) {
                TreeNode parent = nodeMap.get(parentId);
                parent.addChild(node);
            }
        }

        // 打印树结构
        for (TreeNode node : nodeMap.values()) {
            if (node.isRoot()) {
                node.print();
            }
        }
    }

    static class TreeNode {
        int id;
        String name;
        TreeNode parent;
        List<TreeNode> children = new ArrayList<>();

        TreeNode(int id, String name) {
            this.id = id;
            this.name = name;
        }

        void addChild(TreeNode child) {
            children.add(child);
            child.parent = this;
        }

        boolean isRoot() {
            return parent == null;
        }

        void print() {
            print(0);
        }

        void print(int level) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < level; i++) {
                sb.append("  ");
            }
            sb.append(name);
            System.out.println(sb.toString());
            for (TreeNode child : children) {
                child.print(level + 1);
            }
        }
    }
}

示例解析

上面的代码示例中，我们使用了一个内部类TreeNode来表示树中的节点。每个TreeNode对象都有一个唯一的ID和一个名称。它还包含一个父节点的引用和一个子节点列表。

在main方法中，我们首先创建了一个JSONArray对象，并将其作为输入数据。然后，我们创建了一个Map<Integer, TreeNode>用于存储节点ID和对应的TreeNode对象。

接下来，我们遍历JSONArray，提取每个JSON对象的ID、名称和父节点ID。然后，我们创建一个TreeNode对象，并将其存储到nodeMap中。

如果节点的父节点ID不为0（即存在父节点），我们从nodeMap中获取父节点并将当前节点添加到父节点的子节点列表中