java中将id,pid的树节点数据转换为children层级格式的数据 ...

import java.util.*;
// 定义树节点类
class TreeNode {
int id;
int pid;
String name;
List children;

1. public TreeNode(int id, int pid, String name) {
2.     this.id = id;
3.     this.pid = pid;
4.     this.name = name;
5.     this.children = new ArrayList<>();
6. }

复制代码

}
public class TreeNodeConverter {
public static List convertToTree(List nodes) {
// 存储节点映射
Map<Integer, TreeNode> nodeMap = new HashMap<>();
// 存储根节点
List rootNodes = new ArrayList<>();

1.     // 首先将所有节点放入映射中
2.     for (TreeNode node : nodes) {
3.         nodeMap.put(node.id, node);
4.     }
6.     // 构建树结构
7.     for (TreeNode node : nodes) {
8.         int pid = node.pid;
9.         if (pid == 0) {
10.             // 如果父节点 ID 为 0，说明是根节点
11.             rootNodes.add(node);
12.         } else {
13.             // 找到父节点
14.             TreeNode parent = nodeMap.get(pid);
15.             if (parent != null) {
16.                 parent.children.add(node);
17.             }
18.         }
19.     }
21.     return rootNodes;
22. }
24. public static void main(String[] args) {
25.     List<TreeNode> nodes = new ArrayList<>();
26.     nodes.add(new TreeNode(1, 0, "根节点"));
27.     nodes.add(new TreeNode(2, 1, "子节点1"));
28.     nodes.add(new TreeNode(3, 1, "子节点2"));
29.     nodes.add(new TreeNode(4, 2, "孙节点1"));
30.     nodes.add(new TreeNode(5, 2, "孙节点2"));
31.     nodes.add(new TreeNode(6, 3, "孙节点3"));
32.     nodes.add(new TreeNode(7, 3, "孙节点4"));
33.     nodes.add(new TreeNode(8, 4, "曾孙节点1"));
34.     nodes.add(new TreeNode(9, 4, "曾孙节点2"));
35.     nodes.add(new TreeNode(10, 5, "曾孙节点3"));
37.     List<TreeNode> tree = convertToTree(nodes);
39.     // 打印树结构
40.     printTree(tree, 0);
41. }
43. private static void printTree(List<TreeNode> nodes, int level) {
44.     for (TreeNode node : nodes) {
45.         StringBuilder indent = new StringBuilder();
46.         for (int i = 0; i < level; i++) {
47.             indent.append("  ");
48.         }
49.         System.out.println(indent + node.name);
50.         printTree(node.children, level + 1);
51.     }
52. }

复制代码

}

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。