java递归实现树结构

程序员奋斗者GO

于 2021-10-23 21:29:58 发布

阅读量359

点赞数

分类专栏： java

本文链接：https://blog.csdn.net/qq_37896260/article/details/120926635

版权

Java 树形结构递归实体对象数据组织

关键词由CSDN通过智能技术生成

java 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

package com.yan.entity;

import java.util.List;
//实体对象
public class SoftwareTree {
    private Integer id;
    private String name;
    private Integer pid;
    private List<SoftwareTree> children;
}

package com.yan.utils;

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

import com.yan.entity.SoftwareTree;
//递归树封装
public class MenuTree {
    private List<SoftwareTree> SoftwareTreeList = new ArrayList<SoftwareTree>();
    public MenuTree(List<SoftwareTree> SoftwareTreeList) {
        this.SoftwareTreeList=SoftwareTreeList;
    }

    //建立树形结构
    public List<SoftwareTree> builTree(){
        List<SoftwareTree> treeSoftwareTrees =new  ArrayList<SoftwareTree>();
        for(SoftwareTree SoftwareTreeNode : getRootNode()) {
            SoftwareTreeNode=buildChilTree(SoftwareTreeNode);
            treeSoftwareTrees.add(SoftwareTreeNode);
        }
        return treeSoftwareTrees;
    }

    //递归，建立子树形结构
    private SoftwareTree buildChilTree(SoftwareTree pNode){
        List<SoftwareTree> chilSoftwareTrees =new  ArrayList<SoftwareTree>();
        for(SoftwareTree SoftwareTreeNode : SoftwareTreeList) {
            if(SoftwareTreeNode.getPid() == pNode.getId()) {
                chilSoftwareTrees.add(buildChilTree(SoftwareTreeNode));
            }
        }
        pNode.setChildren(chilSoftwareTrees);
        return pNode;
    }

    //获取根节点
    private List<SoftwareTree> getRootNode() {
        List<SoftwareTree> rootSoftwareTreeLists =new  ArrayList<SoftwareTree>();
        for(SoftwareTree SoftwareTreeNode : SoftwareTreeList) {
            if(SoftwareTreeNode.getPid() == 0) {
                rootSoftwareTreeLists.add(SoftwareTreeNode);
            }
        }
        return rootSoftwareTreeLists;
    }
}

//调用
public List<SoftwareTree> getTree(SoftwareTree s) throws ParseException{
		List<SoftwareTree> list = borrowingDao.getByPage(s);

		 MenuTree menuTree =new MenuTree(list);
		 List<SoftwareTree> builTree = menuTree.builTree();
		 
		return builTree;
}