Java實現(xiàn)遞歸查詢樹結構的示例代碼
我們在實際開發(fā)中,肯定會用到樹結構,如部門樹、菜單樹等等。Java后臺利用遞歸思路進行構建樹形結構數(shù)據(jù),返回給前端,能以下拉菜單等形式進行展示。今天,咱們就來說說怎么樣將List集合轉換成TreeList。
一、jar依賴
為了簡化代碼,引入Lombok的Jar包,可省略實體類set()、get()方法。
<dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <version>1.16.12</version> </dependency>
二、樹節(jié)點數(shù)據(jù)類
/** * TreeNode 樹節(jié)點 (定義每一個節(jié)點的信息,即每一個節(jié)點對應一條數(shù)據(jù)信息) */ @Data public class TreeNode { /** 節(jié)點ID */ private Integer id; /** 父節(jié)點ID:頂級節(jié)點為0 */ private Integer parentId; /** 節(jié)點名稱 */ private String label; /** 子節(jié)點 */ private List<TreeNode> children; public TreeNode(Integer id, Integer parentId, String label) { this.id = id; this.parentId = parentId; this.label = label; } }
三、構建樹形類
理解思路(個人):
1、首先獲取所有的根節(jié)點(頂級節(jié)點),即根節(jié)點的parentId = 0。
2、根據(jù)每一個根節(jié)點,與所有節(jié)點集合(數(shù)據(jù))進行判斷,當前節(jié)點是否為其下的子節(jié)點。
3、若是,則遞歸調(diào)用構建樹形;若不是,則表明該節(jié)點不屬于其下子節(jié)點。
4、應繼續(xù)循環(huán)判斷節(jié)點父子關系,直到所有節(jié)點與根節(jié)點判斷完畢。
/** * BuildTree 構建樹形結構 */ public class TreeBuild { // 保存參與構建樹形的所有數(shù)據(jù)(通常數(shù)據(jù)庫查詢結果) public List<TreeNode> nodeList = new ArrayList<>(); /** * 構造方法 * @param nodeList 將數(shù)據(jù)集合賦值給nodeList,即所有數(shù)據(jù)作為所有節(jié)點。 */ public TreeBuild(List<TreeNode> nodeList){ this.nodeList = nodeList; } /** * 獲取需構建的所有根節(jié)點(頂級節(jié)點) "0" * @return 所有根節(jié)點List集合 */ public List<TreeNode> getRootNode(){ // 保存所有根節(jié)點(所有根節(jié)點的數(shù)據(jù)) List<TreeNode> rootNodeList = new ArrayList<>(); // treeNode:查詢出的每一條數(shù)據(jù)(節(jié)點) for (TreeNode treeNode : nodeList){ // 判斷當前節(jié)點是否為根節(jié)點,此處注意:若parentId類型是String,則要采用equals()方法判斷。 if (0 == treeNode.getParentId()) { // 是,添加 rootNodeList.add(treeNode); } } return rootNodeList; } /** * 根據(jù)每一個頂級節(jié)點(根節(jié)點)進行構建樹形結構 * @return 構建整棵樹 */ public List<TreeNode> buildTree(){ // treeNodes:保存一個頂級節(jié)點所構建出來的完整樹形 List<TreeNode> treeNodes = new ArrayList<TreeNode>(); // getRootNode():獲取所有的根節(jié)點 for (TreeNode treeRootNode : getRootNode()) { // 將頂級節(jié)點進行構建子樹 treeRootNode = buildChildTree(treeRootNode); // 完成一個頂級節(jié)點所構建的樹形,增加進來 treeNodes.add(treeRootNode); } return treeNodes; } /** * 遞歸-----構建子樹形結構 * @param pNode 根節(jié)點(頂級節(jié)點) * @return 整棵樹 */ public TreeNode buildChildTree(TreeNode pNode){ List<TreeNode> childTree = new ArrayList<TreeNode>(); // nodeList:所有節(jié)點集合(所有數(shù)據(jù)) for (TreeNode treeNode : nodeList) { // 判斷當前節(jié)點的父節(jié)點ID是否等于根節(jié)點的ID,即當前節(jié)點為其下的子節(jié)點 if (treeNode.getParentId().equals(pNode.getId())) { // 再遞歸進行判斷當前節(jié)點的情況,調(diào)用自身方法 childTree.add(buildChildTree(treeNode)); } } // for循環(huán)結束,即節(jié)點下沒有任何節(jié)點,樹形構建結束,設置樹結果 pNode.setChildren(childTree); return pNode; } }
四、測試案例
/** * TreeController 樹控制層 * 方式:傳遞所有數(shù)據(jù)集合作為參數(shù),調(diào)用buildTree()構建樹形。 */ @RestController @RequestMapping("/tree") public class TreeController { @GetMapping("/treeTest") public AjaxResult treeTest(){ // 模擬測試數(shù)據(jù)(通常為數(shù)據(jù)庫的查詢結果) List<TreeNode> treeNodeList = new ArrayList<>(); treeNodeList.add(new TreeNode(1,0,"頂級節(jié)點A")); treeNodeList.add(new TreeNode(2,0,"頂級節(jié)點B")); treeNodeList.add(new TreeNode(3,1,"父節(jié)點是A")); treeNodeList.add(new TreeNode(4,2,"父節(jié)點是B")); treeNodeList.add(new TreeNode(5,2,"父節(jié)點是B")); treeNodeList.add(new TreeNode(6,3,"父節(jié)點的ID是3")); // 創(chuàng)建樹形結構(數(shù)據(jù)集合作為參數(shù)) TreeBuild treeBuild = new TreeBuild(treeNodeList); // 原查詢結果轉換樹形結構 treeNodeList = treeBuild.buildTree(); // AjaxResult:個人封裝返回的結果體 return AjaxResult.success("測試數(shù)據(jù)",treeNodeList); } }
結果:
{ "msg”:“ 測試數(shù)據(jù)”, "code": 200, "data": [ { "id": 1, "parentId": 0, "label":"頂級節(jié)點A", "children": [ { "id": 3, "parentId": 1, "label":“ 父節(jié)點是A" "children": [ "id": 6, "parentId": 3, "label":“ 父節(jié)點的ID是3 } ] } ] }, { "id": 2, "parentId": 0, "labe1":“ 頂級節(jié)點B", "children": [{ "id": 4, "parentId": 2, "label":“ 父節(jié)點是B" }, { "id": 5, "parentId": 2, "label":" 父節(jié)點是B } ] } ] }
到此這篇關于Java實現(xiàn)遞歸查詢樹結構的示例代碼的文章就介紹到這了,更多相關Java遞歸查詢樹結構內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
Eclipse2020安裝了最新版本的JDK卻無法打開的問題
這篇文章主要介紹了Eclipse2020安裝了最新版本的JDK卻無法打開,提示版本太老的完美解決方法,本文給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12Java使用jdbc連接MySQL數(shù)據(jù)庫實例分析
這篇文章主要介紹了Java使用jdbc連接MySQL數(shù)據(jù)庫,結合實例形式分析了Java基于jdbc鏈接mysql的相關配置及工具類的定義相關操作技巧,需要的朋友可以參考下2018-07-07springboot mybatis調(diào)用多個數(shù)據(jù)源引發(fā)的錯誤問題
這篇文章主要介紹了springboot mybatis調(diào)用多個數(shù)據(jù)源引發(fā)的錯誤問題,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-01-01SpringBoot Redis批量存取數(shù)據(jù)的操作
這篇文章主要介紹了SpringBoot Redis批量存取數(shù)據(jù)的操作,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2021-08-08