Java創(chuàng)建樹形結(jié)構(gòu)算法實(shí)例代碼

更新時(shí)間：2017年01月20日 16:25:07 作者：路偉

本篇文章主要介紹了Java創(chuàng)建樹形結(jié)構(gòu)算法實(shí)例代碼，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下。

在JavaWeb的相關(guān)開發(fā)中經(jīng)常會(huì)涉及到多級(jí)菜單的展示，為了方便菜單的管理需要使用數(shù)據(jù)庫(kù)進(jìn)行支持，本例采用相關(guān)算法講數(shù)據(jù)庫(kù)中的條形記錄進(jìn)行相關(guān)組裝和排序講菜單組裝成樹形結(jié)構(gòu)。

首先是需要的JavaBean

import java.io.Serializable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

/**
 * <Description> 菜單擴(kuò)展<br>
 */
public class MenuExt implements Serializable {

  /**
   * 菜單ID
   */
  private Long id;

  /**
   * 菜單名稱
   */
  private String name;

  /**
   * 菜單地址
   */
  private String url;

  /**
   * 菜單圖標(biāo)
   */
  private String icon;

  /**
   * 父菜單ID
   */
  private Long parentId;

  /**
   * 子菜單
   */
  private List<MenuExt> children = new ArrayList<MenuExt>();

  /**
   * 菜單順序
   */
  private Integer ordby;

  /**
   * 菜單狀態(tài)
   */
  private String state;

  //省略Getter和Setter


  /**
   * 
   * <Description> 孩子節(jié)點(diǎn)排序<br>
   * 
   */
  public void sortChildren() {
    Collections.sort(children, new Comparator<MenuExt>() {
      @Override
      public int compare(MenuExt menu1, MenuExt menu2) {
        int result = 0;

        Integer ordby1 = menu1.getOrdby();
        Integer ordby2 = menu2.getOrdby();

        Long id1 = menu1.getId();
        Long id2 = menu2.getId();
        if (null != ordby1 && null != ordby2) {
          result = (ordby1 < ordby2 ? -1 : (ordby1 == ordby2 ? 0 : 1));
        } else {
          result = (id1 < id2 ? -1 : (id1 == id2 ? 0 : 1));
        }
        return result;
      }

    });
    // 對(duì)每個(gè)節(jié)點(diǎn)的下一層節(jié)點(diǎn)進(jìn)行排序
    for (Iterator<MenuExt> it = children.iterator(); it.hasNext();) {
      it.next().sortChildren();
    }
  }

  public List<MenuExt> getChildren() {
    return children;
  }

  public void setChildren(List<MenuExt> children) {
    this.children = children;
  }
}

Java算法

public static List<MenuExt> createTreeMenus(List<MenuExt> menus) {
    List<MenuExt> treeMenus = null;
    if (null != menus && !menus.isEmpty()) {
      // 創(chuàng)建根節(jié)點(diǎn)
      MenuExt root = new MenuExt();
      root.setName("菜單根目錄");

      // 組裝Map數(shù)據(jù)
      Map<Long, MenuExt> dataMap = new HashMap<Long, MenuExt>();
      for (MenuExt menu : menus) {
        dataMap.put(menu.getId(), menu);
      }

      // 組裝樹形結(jié)構(gòu)
      Set<Entry<Long, MenuExt>> entrySet = dataMap.entrySet();
      for (Entry<Long, MenuExt> entry : entrySet) {
        MenuExt menu = entry.getValue();
        if (null == menu.getParentId() || 0 == menu.getParentId()) {
          root.getChildren().add(menu);
        } else {
          dataMap.get(menu.getParentId()).getChildren().add(menu);
        }
      }

      // 對(duì)樹形結(jié)構(gòu)進(jìn)行二叉樹排序
      root.sortChildren();
      treeMenus = root.getChildren();
    }
    return treeMenus;
  }

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

您可能感興趣的文章: