java树生成

本文介绍了一种基于Java实现的树形数据结构,包括抽象基类TreeNode、具体节点类CostCenterNode及辅助类TreeHelper。TreeNode定义了树节点的基本属性与方法;CostCenterNode实现了特定业务场景下的树节点,并支持排序;TreeHelper提供了生成树形结构的功能。

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

TreeNode基类

/**
 * 树节点
 */
public abstract class TreeNode {   
    private String id;
    private String parentId;
    private String name;
    private List children;

    /**
     * 是否为根节点
     * 根据当前节点的parentId判断该节点是否为根节点
     *
     * @param parentId
     * @return
     */
    public abstract boolean isRoot(String parentId);

    /**
     * 子节点排序
     *
     * @param children
     */
    public abstract void sortChildren(List children);


    public void addChildNode(TreeNode treeNode) {
        initChildList();
        children.add(treeNode);
    }

    public void initChildList() {
        if (children == null)
            children = new ArrayList<>();
    }

    // getter setter
}

TreeHelper

/**
 * 树生成helper
 */
public class TreeHelper {


    public static <T extends TreeNode> void generateTree(Map<String, T> nodeMap) {
        // 建立父子关系
        for (TreeNode node : nodeMap.values()) {
            String parentId = node.getParentId();

            if (node.isRoot(parentId)) {
                continue;
            }
            if (nodeMap.containsKey(parentId)) {
                TreeNode parentNode = nodeMap.get(parentId);
                if (parentNode == null) {
                    throw new InternalException("生成树失败");
                } else {
                    parentNode.addChildNode(node);
                }
            }
        }

        // sort
        for (T t : nodeMap.values()) {
            List children = t.getChildren();
            if (children != null && children.size() > 0) {
                t.sortChildren(children);
            }
        }

    }


}

/**
 * 具体树节点
 */
public class CostCenterNode extends TreeNode implements Comparable<CostCenterNode>{
    private String code;
    private int position;
    private int employeeCount;

    @Override
    public boolean isRoot(String parentId) {
        if (StringUtil.isEmpty(parentId)) {
            return true;
        }
        return false;
    }

    @Override
    public void sortChildren(List children) {
        Collections.sort(children);
    }

    public CostCenterNode(CostCenterDO centerDO) {
        this.setId(centerDO.getCostId());
        this.setParentId(centerDO.getParentId());
        this.setName(centerDO.getName());
        this.setCode(centerDO.getCode());
        this.setPosition(centerDO.getPosition());
        this.setChildren(new ArrayList());
    }

    // getter setter

    @Override
    public int compareTo(CostCenterNode o) {
        // 按照position升序排列
        if (this.position > o.getPosition()) {
            return 1;
        } else {
            return -1;
        }
    }
}
版权声明:本文为博主原创文章,转载请注明出处。 https://blog.youkuaiyun.com/fgszdgbzdb/article/details/68947092
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值