List转为Tree的方法

package com.xicheng.test;

import java.util.List;

/**
 * 2018年3月26日上午7:32:37
 * 树节点实体类
 */
public class TreeNode {
	private String id;
	private String parentId;
	private String name;
	private List<TreeNode> children;
	public String getId() {
		return id;
	}
	public void setId(String id) {
		this.id = id;
	}
	public String getParentId() {
		return parentId;
	}
	public void setParentId(String parentId) {
		this.parentId = parentId;
	}
	public String getName() {
		return name;
	}
	public void setName(String name) {
		this.name = name;
	}
	public List<TreeNode> getChildren() {
		return children;
	}
	public void setChildren(List<TreeNode> children) {
		this.children = children;
	}
	@Override
	public String toString() {
		return "TreeNode [id=" + id + ", parentId=" + parentId + ", name=" + name + ", children=" + children + "]";
	}
	public TreeNode() {
	}
	public TreeNode(String id, String name, String parentId) {
		this.id = id;
		this.parentId = parentId;
		this.name = name;
	}
}

package com.xicheng.test;

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

public class List2Tree {
	public static void main(String[] args) {
		TreeNode t1 = new TreeNode("1","p1","0");
		TreeNode t2 = new TreeNode("2","p2","0");
		TreeNode t3 = new TreeNode("3","s11","1");
		TreeNode t4 = new TreeNode("4","s12","1");
		TreeNode t5 = new TreeNode("5","s21","2");
		TreeNode t6 = new TreeNode("6","s22","2");
		TreeNode t7 = new TreeNode("7","s23","2");
		
		List<TreeNode> treeList = new ArrayList<>();
		Collections.addAll(treeList, t1,t2,t3,t4,t5,t6,t7);
		
		
		System.out.println(toTree01(treeList));
		System.out.println(toTree02(treeList));
	}
	
	/**
	 * 方式一：使用两层循环实现
	 * @param list
	 * @return
	 */
	public static List<TreeNode> toTree01(List<TreeNode> treeList) {
		List<TreeNode> retList = new ArrayList<>();
		
		for (TreeNode parent : treeList) {
			if ("0".equals(parent.getParentId())) {
				retList.add(parent);
			}
			
			for (TreeNode child : treeList) {
				if (child.getParentId() == parent.getId()) {
					if (parent.getChildren() == null) {
						parent.setChildren(new ArrayList<>());
					}
					parent.getChildren().add(child);
				}
			}
		}
		return retList;
	}
	
	/**
	 * 方式二：使用递归实现
	 * @param treeList
	 * @return
	 */
	public static List<TreeNode> toTree02(List<TreeNode> treeList) {
		List<TreeNode> retList = new ArrayList<>();
		for (TreeNode parent : treeList) {
			if ("0".equals(parent.getParentId())) {
				retList.add(findChildren(parent, treeList));
			}
		}
		return retList;
	}

	private static TreeNode findChildren(TreeNode parent, List<TreeNode> treeList) {
		for (TreeNode child : treeList) {
			if (parent.getId().equals(child.getParentId())) {
				if (parent.getChildren() == null) {
					parent.setChildren(new ArrayList<>());
				}
				parent.getChildren().add(findChildren(child, treeList));
			}
		}
		return parent;
	}
}