java中把树转成list_JAVA 把一个List转成树形结构

package com.stu;

import java.util.ArrayList;

import java.util.List;

public class TreeUtils {

//把一个List转成树

static List buidTree(List list){

List tree=new ArrayList<>();

for(Node node:list){

if(node.getPid() == 0){

tree.add(findChild(node,list));

}

}

return tree;

}

static Node findChild(Node node, List list){

for(Node n:list){

if(n.getPid() == node.getId()){

if(node.getChildren() == null){

node.setChildren(new ArrayList());

}

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 list=new ArrayList<>();

list.add(node1);

list.add(node2);

list.add(node3);

list.add(node4);

list.add(node5);

list.add(node6);

List nodes = buidTree(list);

System.out.println(nodes);

}

}