快捷導(dǎo)航

java TreeUtil菜單遞歸工具類

更新時間：2020年08月26日 16:50:21 作者：男人要霸氣

這篇文章主要為大家詳細介紹了java TreeUtil菜單遞歸工具類，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

本文實例為大家分享了java TreeUtil菜單遞歸工具類的具體代碼，供大家參考，具體內(nèi)容如下

package com.admin.manager.storeService.util;

import com.admin.manager.storeService.entity.Menu;

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

/**
 * @author m
 * @date 2019/12/16
 */
public class TreeUtil {

 /**
 * 獲取菜單樹
 * @param menus 所有菜單
 * @return
 */
 public static List<Menu> menuTree(List<Menu> menus) {
 //返回的菜單樹
 List<Menu> rootMenus = new ArrayList<>();
 for (Menu menu : menus) {
  //pid（上級Id）為0的是根菜單
  if ("0".equals(menu.getPid())) {
  rootMenus.add(menu);
  }
 }
 //遍歷，找到二級菜單（根菜單的id和所有菜單中的pid比較）
 for (Menu rootMenu : rootMenus) {
  List<Menu> child = getChild(String.valueOf(rootMenu.getId()), menus);
  rootMenu.setChildren(child);
 }
 return rootMenus;
 }


 /**
 * 遞歸獲取下級菜單
 * @param pid 上級Id
 * @param menus 所有菜單
 * @return
 */
 public static List<Menu> getChild(String pid, List<Menu> menus) {
 //子菜單列表
 List<Menu> childList = new ArrayList<>();
 for (Menu menu : menus) {
  if (pid.equals(menu.getPid())) {
  childList.add(menu);
  }
 }
 //遍歷 遞歸獲取子菜單的子菜單
 for (Menu menu : childList) {
  List<Menu> child = getChild(String.valueOf(menu.getId()), menus);
  menu.setChildren(child);
 }
 //遞歸出口 childList長度為0
 if (childList.size() == 0) {
  return new ArrayList<>();
 }
 return childList;
 }
}

上面這種遞歸，如果遇到大數(shù)據(jù)量，效率是極低的。比如前段時間剛實現(xiàn)了省市區(qū)鎮(zhèn)四層的樹結(jié)構(gòu)，面對5萬條的數(shù)據(jù)，竟然120s才遞歸完，這也太慢了，將來需求變成了省市區(qū)鎮(zhèn)鄉(xiāng)五層的結(jié)構(gòu)，那不就掛了？

大眼一看，不是跟上面的一樣嘛。仔細看 it.remove();

假設(shè)有五萬的數(shù)據(jù)，根節(jié)點1萬，二級節(jié)點1萬，三級節(jié)點1萬，4級節(jié)點2萬。如果按照上面的邏輯走，每次都要從5萬中數(shù)據(jù)遞歸查找；而下面的邏輯是找到N級節(jié)點并且在總list中刪除。假設(shè)找到根節(jié)點的1萬數(shù)據(jù)，并且把他們從總list中刪除，總list剩4萬條數(shù)據(jù)，依次這樣，到了找4級節(jié)點時，總list就已經(jīng)剩2萬條數(shù)據(jù)了，總數(shù)據(jù)量從5萬變成了2萬，循環(huán)次數(shù)就少了，效率也就高了

import com.energy.service.vo.RegionTree;

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

/**
 * @author m
 * @date 2020/4/2
 */
public class TreeUtil {
 /**
 * 創(chuàng)建樹結(jié)構(gòu)
 * @param regionTrees
 * @return
 */
 private List<RegionTree> tree(List<RegionTree> regionTrees) {
 //返回的節(jié)點樹
 List<RegionTree> rootNodes = new ArrayList<>();
 Iterator<RegionTree> it = regionTrees.iterator();
 while (it.hasNext()) {
  RegionTree next = it.next();
  //parent（上級Id）為0的是根節(jié)點
  if ("0".equals(next.getParentId())) {
  rootNodes.add(next);
  it.remove();
  }
 }
 //遍歷，找到二級節(jié)點
 for (RegionTree regionTree : rootNodes) {
  List<RegionTree> child = getChild(regionTrees, regionTree.getId().toString());
  regionTree.setChildren(child);
 }
 return rootNodes;

 }

 /**
 * 查子節(jié)點
 *
 * @param regions
 * @param parentId
 * @return
 */
 private List<RegionTree> getChild(List<RegionTree> regions, String parentId) {

 //子節(jié)點列表
 List<RegionTree> childList = new ArrayList<>();
 Iterator<RegionTree> it = regions.iterator();
 while (it.hasNext()) {
  RegionTree regionTree = it.next();
  if (parentId.equals(regionTree.getParentId())) {
  childList.add(regionTree);
  it.remove();
  }
 }

 //遍歷 遞歸獲取子節(jié)點的子節(jié)點
 for (RegionTree regionTree : childList) {
  List<RegionTree> child = getChild(regions, regionTree.getId().toString());
  regionTree.setChildren(child);
 }
 //遞歸出口 childList長度為0
 if (childList.size() == 0) {
  return new ArrayList<>();
 }
 return childList;
 }
}

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: