使用java构建树形结构的多种实现工具类

whale-52赫兹

已于 2023-11-20 16:31:23 修改

阅读量762

点赞数

分类专栏：笔记数据结构文章标签： java linux 开发语言

于 2023-11-20 15:33:00 首次发布

本文链接：https://blog.csdn.net/weixin_42645678/article/details/134510365

版权

笔记同时被 2 个专栏收录

12 篇文章

订阅专栏

数据结构

2 篇文章

订阅专栏

本文介绍了Java中用于构建树形结构的工具类，包括TreeNode基类的定义和TreeUtil中的三种实现方法：celerityBuild（双循环）、build（嵌套循环）和buildByRecursive（迭代构建），展示了如何使用这些工具来组织节点关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

比较通用的一套树形结构构建工具类。

1.属性节点的基类，包含三个主要属性，当前节点id、父节点id、子节点集合

import lombok.Data;

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

/**
 * 树结构节点
 * @Author dong
 * @Date 2023/11/20 下午 3:27
 */
@Data
public class TreeNode<T extends TreeNode> {
    /**
     * 当前节点id
     */
    protected Object id;
    /**
     * 父节点id
     */
    protected Object parentId;
    /**
     * 子节点列表
     */
    protected List<T> children;

    public void add(T node) {
        if(children == null){
            children = new ArrayList<>();
        }
        children.add(node);
    }
}

2.构建树形结构的工具类，总共三个实现，参数都是节点集合+根节点id。
三个实现直接看代码吧。个人比较喜欢用的是一个celerityBuild，n*2即可构建出树

import lombok.experimental.UtilityClass;

import java.util.*;

/**
 * 树工具类
 * @Author dong
 * @Date 2023/11/20 下午 3:29
 */
@UtilityClass
public class TreeUtil {

	/**
     * 双循环构建树 n*2
     * @Author dong
     * @Date 2023/11/20 下午 3:25
     * @param treeNodes
     * @param root
     * @return java.util.List<T>
     */
    public <T extends TreeNode> List<T> celerityBuild(List<T> treeNodes, Object root) {
        List<T> result = new ArrayList<>();
        Map<Object, TreeNode> nodeMap = new HashMap<>();
        for (T treeNode : treeNodes) {
            nodeMap.put(treeNode.getId(),treeNode);
        }

        for (T treeNode : treeNodes) {
            if (Objects.equals(root, treeNode.getParentId())) {
                result.add(treeNode);
            }
            TreeNode p = nodeMap.get(treeNode.getParentId());
            if(p!=null){
                p.add(treeNode);
            }
        }
        return result;
    }

    /**
     * 嵌套循环构建树 n*n
     * @Author dong
     * @Date 2023/11/20 下午 3:26
     * @param treeNodes
     * @param root
     * @return java.util.List<T>
     */
    public <T extends TreeNode> List<T> build(List<T> treeNodes, Object root) {
        List<T> result = new ArrayList<>();
        for (T treeNode : treeNodes) {
            if (Objects.equals(root, treeNode.getParentId())) {
                result.add(treeNode);
            }
            for (T it : treeNodes) {
                if (Objects.equals(it.getParentId(), treeNode.getId())) {
                    treeNode.add(it);
                }
            }
        }
        return result;
    }


    /**
     * 迭代构建树
     * @Author dong
     * @Date 2023/11/20 下午 3:25
     * @param treeNodes
     * @param root
     * @return java.util.List<T>
     */
    public <T extends TreeNode> List<T> buildByRecursive(List<T> treeNodes, Object root) {
        List<T> result = new ArrayList<T>();
        for (T treeNode : treeNodes) {
            if (Objects.equals(root, treeNode.getParentId())) {
                result.add(loadChildren(treeNode, treeNodes));
            }
        }
        return result;
    }

    private  <T extends TreeNode> T loadChildren(T treeNode, List<T> treeNodes) {
        for (T it : treeNodes) {
            if (treeNode.getId() == it.getParentId()) {
                treeNode.add(loadChildren(it, treeNodes));
            }
        }
        return treeNode;
    }
    
}