java不需要递归列表转树形结构

有时候我们需要将列表结构的数据转成树形结构的数据

废话不多说直接上代码

基础类

`@Data

public class TreeNode {

private Long id;

private Long parentId;

private List<TreeNode> childrenList;

}工具类import org.apache.commons.lang3.StringUtils;

import org.springframework.util.CollectionUtils;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Objects;

import java.util.stream.Collectors;

/**

• 树形结构工具类.

• @date: 2020-5-8

• @version: 1.0

• @author: liyanglin
  
  */
  
  public class TreeNodeUtil {

  /**

  • 将列表转换成树。

  • @param treeNodeList 列表数据

  • @param rootId 根节点id

  • @param rootType 根节点id类型 1-将rootId与id值相等的作为根节点，2-将rootId与parentId值相等的作为根节点

  • @param

  • @return 返回树结构数据
    
    */
    
    public static List listToTree(List treeNodeList, Long rootId, int rootType) {
    
    // 储存根节点集合
    
    List baseNodes = new ArrayList<>();
    
    // 节点id->对应的节点
    
    HashMap<Long, TreeNode> sonMap = new HashMap<>(baseNodes.size());
    
    // 节点id->节点对应的子节点
    
    HashMap<Long, List> parentMap = new HashMap<>(baseNodes.size());

    for (T currentNode : treeNodeList) {
    
    Long currentId = currentNode.getId();
    
    Long parentId = currentNode.getParentId();

     // 把节点先放在Map中,为了以后获取当前节点的父节点
     sonMap.put(currentId, currentNode);
    
     if (parentMap.get(parentId) == null) {
         parentMap.put(parentId, new ArrayList<>());
     }
     //将子节点加入父节点
     parentMap.get(parentId).add(currentNode);
    
     // 判断当前节点是否有父节点
     if (sonMap.get(parentId) != null) {
         if (sonMap.get(parentId).getChildrenList() == null) {
             sonMap.get(parentId).setChildrenList(new ArrayList<>());
         }
         // 将当前节点添加到父节点中
         sonMap.get(parentId).getChildrenList().add(currentNode);
     }
    
     // 判断当前节点是否有子节点
     if (!CollectionUtils.isEmpty(parentMap.get(currentId))) {
         //将所有的子节点加入到当前节点
         currentNode.setChildrenList(parentMap.get(currentId));
     }
    
     //根据节点id过滤根节点
     if (rootType == 1 && currentNode.getId().longValue() == rootId.longValue()) {
         baseNodes.add(currentNode);
     }
    
     //根据父节点id过滤根节点
     if (rootType == 2 && currentNode.getParentId().longValue() == rootId.longValue()) {
         baseNodes.add(currentNode);
     }
    

    }
    
    return baseNodes;
    
    }
    
    』测试类public class TreeToListTest {
    
    @Test
    
    public void test() throws Exception {

    List treeNodes=new ArrayList<>();
    
    treeNodes.add(createNode(1L,0L));
    
    treeNodes.add(createNode(2L,10L));
    
    treeNodes.add(createNode(3L,11L));
    
    treeNodes.add(createNode(4L,3L));
    
    treeNodes.add(createNode(5L,3L));
    
    treeNodes.add(createNode(6L,31L));
    
    treeNodes.add(createNode(7L,13L));
    
    treeNodes.add(createNode(8L,2L));
    
    treeNodes.add(createNode(9L,2L));
    
    treeNodes.add(createNode(10L,1L));
    
    treeNodes.add(createNode(11L,1L));
    
    treeNodes.add(createNode(12L,1L));
    
    treeNodes.add(createNode(13L,1L));

    List treeNodeList1=TreeNodeUtil.listToTree(treeNodes,1L,1);
    
    }
    
    』`