JAVA 把一个List转成树形结构

package com.stu;

import java.util.ArrayList;
import java.util.List;

public class TreeUtils {

    //把一个List转成树
    static List<Node> buidTree(List<Node> list){
        List<Node> tree=new ArrayList<>();
        for(Node node:list){
            if(node.getPid() == 0){
                tree.add(findChild(node,list));
            }
        }
        return tree;
    }

    static Node findChild(Node node, List<Node> list){
        for(Node n:list){
            if(n.getPid() == node.getId()){
                if(node.getChildren() == null){
                    node.setChildren(new ArrayList<Node>());
                }
                node.getChildren().add(findChild(n,list));
            }
        }
        return node;
    }

    public static void main(String[] args) {
        Node node1=new Node(1,"山东省",0);
        Node node2=new Node(2,"德州市",1);
        Node node3=new Node(3,"夏津县",2);
        Node node4=new Node(4,"济南市",1);
        Node node5=new Node(5,"天津",0);
        Node node6=new Node(6,"红桥区",5);
        List<Node> list=new ArrayList<>();
        list.add(node1);
        list.add(node2);
        list.add(node3);
        list.add(node4);
        list.add(node5);
        list.add(node6);
        List<Node> nodes = buidTree(list);
        System.out.println(nodes);
    }
}

 

在使用Java一个List转换为树形结构时,需要考虑以下几个步骤: 1. 定义树节点类,包含节点ID、父节点ID、节点名称和子节点列表。 2. 遍历List,将每个节点转换为树节点对象,并将其放入一个HashMap中,以便后面查找父节点。 3. 遍历List,将每个节点与其父节点关联起来,将其添加到父节点的子节点列表中。 4. 最后,找到根节点,返回整棵树。 以下是一个示例代码,可以将一个List转换为树形结构: ```java class TreeNode { String id; String parentId; String name; List<TreeNode> children; public TreeNode(String id, String parentId, String name) { this.id = id; this.parentId = parentId; this.name = name; this.children = new ArrayList<>(); } } public class TreeBuilder { public static TreeNode buildTree(List<TreeNode> nodes) { if (nodes == null || nodes.isEmpty()) { return null; } // 将所有节点放入HashMap中,以便查找父节点 Map<String, TreeNode> nodeMap = new HashMap<>(); for (TreeNode node : nodes) { nodeMap.put(node.id, node); } // 遍历所有节点,将每个节点添加到其父节点的子节点列表中 for (TreeNode node : nodes) { if (node.parentId == null || node.parentId.isEmpty()) { // 根节点 continue; } TreeNode parent = nodeMap.get(node.parentId); if (parent != null) { parent.children.add(node); } } // 找到根节点并返回 for (TreeNode node : nodes) { if (node.parentId == null || node.parentId.isEmpty()) { return node; } } return null; } } ``` 使用方法如下: ```java List<TreeNode> nodes = new ArrayList<>(); // 添加节点到列表中 TreeNode root = TreeBuilder.buildTree(nodes); ```
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值