轻松编写高效工具类：构建树形结构的秘籍

目录

前言：

1、BuildTree 工具类的作用和优势：

2、BuildTree 工具类代码

3、依赖的实体类Treevo

---

前言：

在软件开发中，树形结构是一种常见且重要的数据结构。它以树的形式组织数据，其中每个节点可以包含任意数量的子节点。树形结构广泛应用于各个领域，如文件系统、图像处理、网站导航等。

---

树形结构的主要优势在于能够以递归方式组织和访问数据。它提供了一种直观、灵活的方式来表示层次关系，并且能够高效地执行插入、删除和搜索操作。树形结构对于处理包含父子关系的数据非常有用，它能够轻松地表示层级结构和组织结构，使得数据操作更加方便和高效。

---

然而，手动构建树形结构往往复杂且繁琐。需要考虑节点的层级关系、递归处理节点和节点列表等问题。为了简化树形结构的构建过程，可以使用一些工具类或库来辅助开发。下面介绍一个名为 BuildTree 的工具类，用于构建树形结构并解决手动构建树形结构的复杂性和繁琐性。

---

1、BuildTree 工具类的作用和优势：

BuildTree 是一个开发中常用的工具类，用于构建树形结构。它封装了一系列方法，使得构建树形结构变得简单和高效。使用 BuildTree 工具类的优势包括：

• 简化开发流程：BuildTree 工具类提供了一种简单直观的方式来构建树形结构，不需要手动编写复杂的逻辑。

• 提高代码可读性和可维护性：通过使用 BuildTree 工具类，代码的意图更加清晰明了，降低了出错的几率，并且简化了后续的维护工作。

2、BuildTree 工具类代码

package com.zking.util;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class BuildTree {

	/**
	 * 默认-1为顶级节点
	 * @param nodes
	 * @param <T>
	 * @return
	 */
	public static <T> TreeVo<T> build(List<TreeVo<T>> nodes) {

		// 如果节点列表为空，则返回 null
		if (nodes == null) {
			return null;
		}
		
		// 创建一个空的顶级节点列表
		List<TreeVo<T>> topNodes = new ArrayList<TreeVo<T>>();

		for (TreeVo<T> children : nodes) {
			String pid = children.getParentId();
			
			// 如果父节点ID为null或者"-1"，则将当前节点视为顶级节点，添加到顶级节点列表中
			if (pid == null || "-1".equals(pid)) {
				topNodes.add(children);

				continue;
			}

			// 遍历节点列表，将当前节点添加到对应的父节点的子节点列表中
			for (TreeVo<T> parent : nodes) {
				String id = parent.getId();
				if (id != null && id.equals(pid)) {
					parent.getChildren().add(children);
					children.setHasParent(true);
					parent.setChildren(true);
					continue;
				}
			}

		}

		TreeVo<T> root = new TreeVo<T>();
		
		// 根据顶级节点列表的大小决定返回结果
		if (topNodes.size() == 1) {
			// 如果顶级节点列表只有一个节点，将其作为根节点返回
			root = topNodes.get(0);
		} else {
			// 如果顶级节点列表为空或包含多个节点，创建一个虚拟的根节点，将顶级节点列表作为其子节点，然后返回根节点
			root.setId("000");
			root.setParentId("-1");
			root.setHasParent(false);
			root.setChildren(true);
			root.setChecked(true);
			root.setChildren(topNodes);
			root.setText("顶级节点");
			Map<String, Object> state = new HashMap<>(16);
			state.put("opened", true);
			root.setState(state);
		}

		return root;
	}

	/**
	 * 指定idParam为顶级节点
	 * @param nodes
	 * @param idParam
	 * @param <T>
	 * @return
	 */
	public static <T> List<TreeVo<T>> buildList(List<TreeVo<T>> nodes, String idParam) {
		// 如果节点列表为空，则返回 null
		if (nodes == null) {
			return null;
		}
		
		// 创建一个空的顶级节点列表
		List<TreeVo<T>> topNodes = new ArrayList<TreeVo<T>>();

		for (TreeVo<T> children : nodes) {

			String pid = children.getParentId();
			
			// 如果父节点ID为null或与idParam相等，则将当前节点视为顶级节点，添加到顶级节点列表中
			if (pid == null || idParam.equals(pid)) {
				topNodes.add(children);

				continue;
			}

			// 遍历节点列表，将当前节点添加到对应的父节点的子节点列表中
			for (TreeVo<T> parent : nodes) {
				String id = parent.getId();
				if (id != null && id.equals(pid)) {
					parent.getChildren().add(children);
					children.setHasParent(true);
					parent.setChildren(true);

					continue;
				}
			}

		}
		return topNodes;
	}

}

3、依赖的实体类Treevo

package com.zking.util;

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


public class TreeVo<T> {
	/**
	 * 节点ID
	 */
	private String id;
	/**
	 * 显示节点文本
	 */
	private String text;
	/**
	 * 节点状态，open closed
	 */
	private Map<String, Object> state;
	/**
	 * 节点是否被选中 true false
	 */
	private boolean checked = false;
	/**
	 * 节点属性
	 */
	private Map<String, Object> attributes;

	/**
	 * 节点的子节点
	 */
	private List<TreeVo<T>> children = new ArrayList<TreeVo<T>>();

	/**
	 * 父ID
	 */
	private String parentId;
	/**
	 * 是否有父节点
	 */
	private boolean hasParent = false;
	/**
	 * 是否有子节点
	 */
	private boolean hasChildren = false;

	public String getId() {
		return id;
	}

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

	public String getText() {
		return text;
	}

	public void setText(String text) {
		this.text = text;
	}

	public Map<String, Object> getState() {
		return state;
	}

	public void setState(Map<String, Object> state) {
		this.state = state;
	}

	public boolean isChecked() {
		return checked;
	}

	public void setChecked(boolean checked) {
		this.checked = checked;
	}

	public Map<String, Object> getAttributes() {
		return attributes;
	}

	public void setAttributes(Map<String, Object> attributes) {
		this.attributes = attributes;
	}

	public List<TreeVo<T>> getChildren() {
		return children;
	}

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

	public boolean isHasParent() {
		return hasParent;
	}

	public void setHasParent(boolean isParent) {
		this.hasParent = isParent;
	}

	public boolean isHasChildren() {
		return hasChildren;
	}

	public void setChildren(boolean isChildren) {
		this.hasChildren = isChildren;
	}

	public String getParentId() {
		return parentId;
	}

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

	public TreeVo(String id, String text, Map<String, Object> state, boolean checked, Map<String, Object> attributes,
                  List<TreeVo<T>> children, boolean isParent, boolean isChildren, String parentID) {
		super();
		this.id = id;
		this.text = text;
		this.state = state;
		this.checked = checked;
		this.attributes = attributes;
		this.children = children;
		this.hasParent = isParent;
		this.hasChildren = isChildren;
		this.parentId = parentID;
	}

	public TreeVo() {
		super();
	}

	@Override
	public String toString() {
		return "TreeVo [id=" + id + ", text=" + text + ", state=" + state + ", checked=" + checked + ", attributes="
				+ attributes + ", children=" + children + ", parentId=" + parentId + ", hasParent=" + hasParent
				+ ", hasChildren=" + hasChildren + "]";
	}

	
	
}

今天的分享就到这里，记得一键三连兄弟们