使用Java8将列表型数据转为树形数据

三丶胖子

已于 2022-10-26 16:12:46 修改

阅读量632

点赞数

CC 4.0 BY-SA版权

分类专栏： Java基础文章标签： java

于 2022-10-26 16:11:29 首次发布

本文链接：https://blog.youkuaiyun.com/xuhf_1988/article/details/127534650

Java基础专栏收录该内容

8 篇文章

订阅专栏

博客展示了Java相关代码，但未给出具体代码内容。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码如下：

import cn.hutool.core.collection.CollectionUtil;
import com.alibaba.fastjson.JSONObject;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import lombok.Data;
import lombok.experimental.Accessors;
import org.junit.Test;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class Demo {

	@Test
	public void test01() {
	    List<Node> nodes = Lists.newArrayList();
	    Node node = new Node();
	    node.setId(1).setName("root").setPid(0);
	    nodes.add(node);
	    node = new Node();
	    node.setId(2).setName("level1-1").setPid(1);
	    nodes.add(node);
	    node = new Node();
	    node.setId(3).setName("level1-2").setPid(1);
	    nodes.add(node);
	
	    node = new Node();
	    node.setId(4).setName("level2-1").setPid(2);
	    nodes.add(node);
	
	    node = new Node();
	    node.setId(5).setName("level2-1").setPid(2);
	    nodes.add(node);
	
	    Map<Integer, List<Node>> nodeMap = nodes.stream().collect(Collectors.groupingBy(Node::getPid));
	
	    Map<Integer, Node> nodeMap2 = nodes.stream().collect(Collectors.toMap(Node::getId, x -> x));
	    for (Integer pid : nodeMap.keySet()) {
	        List<Node> nodeList = nodeMap.get(pid);
	        Node n = nodeMap2.get(pid);
	        if (n != null && CollectionUtil.isNotEmpty(nodeList)) {
	            n.setChildren(nodeList);
	        }
	    }
	
	    List<Node> result =  nodes.stream().filter(x -> x.getPid() == 0).collect(Collectors.toList());
	    System.out.println(JSONObject.toJSONString(result));
	
	}
	
	@Data
	@Accessors(chain = true)
	public static class Node {
	    private int id;
	    private String name;
	    private int pid;
	    private List<Node> children;
	}
}