Java构建树对象

2019独角兽企业重金招聘Python工程师标准>>>

package com.my.tree;

import java.util.List;

public class Tree {

	private String id; // 节点ID
	
	private String parentId; // 节点父ID
	
	private List<Tree> 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 List<Tree> getChildren() {
		return children;
	}

	public void setChildren(List<Tree> children) {
		this.children = children;
	}
	
}

package com.my.tree;

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

public class TreeUtils {

	public static List<Tree> formatTree(List<Tree> list) {

		List<Tree> nodeList = new ArrayList<Tree>();
		for (Tree node1 : list) {
			boolean mark = false;
			for (Tree node2 : list) {
				if (node1.getParentId() != null && node1.getParentId().equals(node2.getId())) {
					mark = true;
					if (node2.getChildren() == null) {
						node2.setChildren(new ArrayList<Tree>());
					}
					node2.getChildren().add(node1);
					break;
				}
			}
			if (!mark) {
				nodeList.add(node1);
			}
		}
		return nodeList;
	}
	
}

 

转载于:https://my.oschina.net/Tasty881/blog/1545276