O(n)算法快速构建树形结构(权限树、组织树、省市区联动树)

最新推荐文章于 2021-09-18 21:34:41 发布

原创

最新推荐文章于 2021-09-18 21:34:41 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种使用O(n)时间复杂度快速构建树形结构的方法，适用于权限树、组织树和省市区联动树的构建。通过HashMap作为辅助数据结构，仅遍历一次数据即可完成树的构建，避免了传统方法中多次遍历导致的O(n²)时间复杂度。文章提供了一个TreeBean.java的例子，并展示了如何根据id和name构建包含子节点的树形结构。

假设数据库查询返回N条记录有父子关系的数据（通过parent_id来关联），例如：SELECT t.id,t.parent_id as pid,t.name /**, t.level **/ FROM tree_table t。应用层获取到数据后如何根据父子关系构建树形结构呢？通常的做法是多次遍历，如第一次遍历筛选出第一级节点（假设是n1条），第二次遍历这n1条一级节点，以每个节点id为父节点再次遍历N条总数据得出每个一级节点的所有二级节点（假设是n2）条，然后再进行类似遍历查找下级节点。假设总级数（或者树的层数）是L，则总遍历次数是N+n1×N+n2×N+n3×N+···+n(L-1)×N≈N²，如果返回的数据是按一级二级三级顺序排序的话，遍历次数会相对少一些，但总体时间复杂度还是O(n²)。

可以使用HashMap来辅助，只需遍历一次即可构建树形结构。具体是，以id为Map的key，name以及子节点list为value，一次遍历下来即可构建关联完所有节点的子节点。

TreeBean.java 对应数据库查询出来的数据。

public class TreeBean {

	private Long id;
	
	private Long parentId;
	
	private String name;

	public Long getId() {
		return id;
	}

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

	public Long getParentId() {
		return parentId;
	}

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

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}
	
	@Override
	public String toString() {
		retur