java从数据库中读取数据-生成菜单树tree之算法优化

  算法1：递归算法


  算法2：循环算法（推荐--优化算法）

public List<ProductType> treeData() {
      // 方案一：通过递归的方式来获得菜单
      return getTreeDataRecursion(0L);
  
}

// 递归方法，调用此方法时候，将 0 传入，0代表一级菜单
private List<ProductType> getTreeDataRecursion(long id) {
      // 获取所有的子级
      Wrapper<ProductType> wrapper = new EntityWrapper<>();
      wrapper.eq("pid", id);
      List<ProductType> productTypes = selectList(wrapper);

      // 判断子级是否还有子级
      if (productTypes == null || productTypes.size() < 1) {
         // 如果没有子级则返回空
         return null;
      }
      // 循环获得的子级，将子级放入父级中
      for (ProductType productType : productTypes) {
         // 递归方法，自己调用自己
         List<ProductType> treeDataRecursion = getTreeDataRecursion(productType.getId());
         // 将子级放入父级中
         productType.setChildren(treeDataRecursion);
      }
      // 返回菜单集合
      return productTypes;
}

 public List<ProductType> treeData() {

      // 方案二：通过循环的方式获得菜单。这里将 0 传入，因为0在此次数据表中代表一级菜单
      List<ProductType> dataLoop = getTreeDataLoop(0L);
      return dataLoop;    
 }

 // 优化算法
 private List<ProductType> getTreeDataLoop(long id) {
      // 新建一个list来存放一级菜单
      List<ProductType> tree = new ArrayList<>();

      Wrapper<ProductType> wrapper = new EntityWrapper<>();
      // 获取所有的菜单数据
      List<ProductType> productTypes = selectList(wrapper);

      Map<Long, ProductType> map = new HashMap<>();
      // 将所有的数据，以键值对的形式装入map中
      for (ProductType productType : productTypes) {
         map.put(productType.getId(), productType);
      }

      for (ProductType productType : productTypes) {
         // 如果id是父级的话就放入tree中
         if (productType.getPid().longValue() == id) {
            tree.add(productType);
         } else {
            // 子级通过父id获取到父级的类型
            ProductType parent = map.get(productType.getPid());
            // 父级获得子级，再将子级放到对应的父级中
            parent.getChildren().add(productType);
         }
      }
      return tree;
 }