简单的列表转树处理算法Java实现

一、引言

公司新入职员工对列表转树的方法不了解，和他沟通了算法还是不明白的样子，网上找了几个类似的例子都不是很满意，只好自己动手写了一个，有需要的童鞋自取。

 

二、解决思路

主要的算法：

1、找出所有的根节点；（没有上级节点或列表中找不到父节点的节点都作为根基点处理）

2、找出每个根节点下的各级子节点；

3、打印输出最后的树形结构；

 

三、源代码

package com.duoduo.demo;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 列表转树
 * @author cgs1999@126.com
 * @date 2018年10月15日 上午11:22:41
 */
public class ListToTreeTest {
    public static void main(String[] args) {
        List<TreeNode<Integer>> nodes = new ArrayList<TreeNode<Integer>>(0);
        nodes.add(new TreeNode<Integer>(1, "TreeNode 1", 0));
        nodes.add(new TreeNode<Integer>(2, "TreeNode 2", 1));
        nodes.add(new TreeNode<Integer>(3, "TreeNode 3", 1));