快速构建树形结构数据(非递归)

 

       通过平铺的原始树形结构数据，快熟构建节点的父子管理，完成树型结构数据的构建，一般在配合前段树形机构显示中有特殊需求的情况中使用。如果需要提出一个父子结构数据中部分数据，如果利用hibernate的级联查询，是很方面获取树形结构的全量，但要剔除部分数据就麻烦了，而且hinernate执行的时候是多个SQL，效率也差一些（不考虑缓存哈）。

 

       原始数据结构：

       ID,PID,NAME....

       目标数据(json)：

       {

            id:1

            name:'NAME1',

            children:[

                {id:2,name:'name2',children:[...]}

            ]

       }

 

好了，直接上代码先。

/**
 * 快速构建树结构
 * 
 * 非递归，使用2次循环完成构建。
 * @author zhangpu
 *
 */
public class QuickTreeBuilder {

	public static void main(String[] args) {
		// 构建原始数据：这里为了简便，原始数据集合使用的也是Node对象，
		// 实际情况可以是其他任意entity，只需要在后面构建树时进行转化就OK
		List<Node> orginals = new ArrayList<Node>();
		// parentId = 0 表示顶层，你也可以构建原始数据parentId=null是顶层，由你决定。
		for (int i = 1; i <= 5; i++) {
			orginals.add(new Node(i, i - 1, "name" + i));
		}
		// 构建一个Map,把所有原始数据的ID作为Key,原始数据对象作为VALUE
		Map<Integer, Node> dtoMap = new HashMap<Integer, Node>();
		for (Node node : orginals) {
			// 如果原始数据对象不是Node,这里就可以直接写个conver转化
			// 原始数据对象为Node，放入dtoMap中。
			dtoMap.put(node.getId(), node);
		}

		List<Node> result = new ArrayList<Node>();
		for (Map.Entry<Integer, Node> entry : dtoMap.entrySet()) {
			Node node = entry.getValue();
			if (node.getParentId() == 0) {
				// 如果是顶层节点，直接添加到结果集合中
				result.add(node);
			} else {
				// 如果不是顶层节点，找的起父节点，然后添加到父节点的子节点中
				if (dtoMap.get(node.getParentId()) != null) {
					dtoMap.get(node.getParentId()).addChild(node);
				}
			}
		}

		// 如果有排序需求，可以在最后对输出的结构进行排序
		// 打印结果数据
		System.out.println(result);

	}

	static class Node {
		private int id;
		private int parentId;
		private String name;
		private List<Node> children = new ArrayList<Node>();

		public Node() {
		}

		public Node(int id, int parentId, String name) {
			super();
			this.id = id;
			this.parentId = parentId;
			this.name = name;
		}

		public void addChild(Node node) {
			this.children.add(node);
		}

		public int getId() {
			return id;
		}

		public void setId(int id) {
			this.id = id;
		}

		public int getParentId() {
			return parentId;
		}

		public void setParentId(int parentId) {
			this.parentId = parentId;
		}

		public String getName() {
			return name;
		}

		public void setName(String name) {
			this.name = name;
		}

		public List<QuickTreeBuilder.Node> getChildren() {
			return children;
		}

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

		@Override
		public String toString() {
			return "{id:" + id + ", parentId:" + parentId + ", name:" + name + ", children:" + children + "}";
		}

	}

}