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

淡淡的烟草味﹌ 2022-03-26 06:22 218阅读 0赞

## 这里介绍两种从数据库总读取数据生成菜单树的算法： ##

**    算法1：递归算法**

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

### **一、数据库设计：** ###

**![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1cHJlemhlbmc_size_16_color_FFFFFF_t_70][]**

**实例中的重要字段：**

![201901152343459.png][]

###  **     二、算法1：递归算法** ###

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;
    }

###  三、算法2：循环算法（这是优化算法） ###

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;
     }

### 三、测试结果 ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1cHJlemhlbmc_size_16_color_FFFFFF_t_70 1][]

![20190115235827293.png][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1cHJlemhlbmc_size_16_color_FFFFFF_t_70]: /images/20220326/5a0abc651fc7454494c29a86eb01cb22.png
[201901152343459.png]: /images/20220326/15739679a927431b8884a2cc1f3a64ba.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3N1cHJlemhlbmc_size_16_color_FFFFFF_t_70 1]: /images/20220326/804642d0524c492d9f6668d8932925d1.png
[20190115235827293.png]: /images/20220326/ddc774ecb56e4130996a1bcc74d76485.png