java将list转为树形结构的方法

ゝ一纸荒年。 2023-02-18 14:23 40阅读 0赞

/** - listToTree - <p>方法说明<p> - 将JSONArray数组转为树状结构 - @param arr 需要转化的数据 - @param id 数据唯一的标识键值 - @param pid 父id唯一标识键值 - @param child 子节点键值 - @return JSONArray */
    public static JSONArray listToTree(JSONArray arr,String id,String pid,String child){ 
       JSONArray r = new JSONArray();
       JSONObject hash = new JSONObject();
       //将数组转为Object的形式，key为数组中的id
       for(int i=0;i<arr.size();i++){ 
      JSONObject json = (JSONObject) arr.get(i);
      hash.put(json.getString(id), json);
       }
       //遍历结果集
       for(int j=0;j<arr.size();j++){ 
      //单条记录
      JSONObject aVal = (JSONObject) arr.get(j);
      //在hash中取出key为单条记录中pid的值
      JSONObject hashVP = (JSONObject) hash.get(aVal.get(pid).toString());
      //如果记录的pid存在，则说明它有父节点，将她添加到孩子节点的集合中
      if(hashVP!=null){ 
         //检查是否有child属性
         if(hashVP.get(child)!=null){ 
            JSONArray ch = (JSONArray) hashVP.get(child);
            ch.add(aVal);
            hashVP.put(child, ch);
         }else{ 
            JSONArray ch = new JSONArray();
            ch.add(aVal);
            hashVP.put(child, ch);
         }
      }else{ 
         r.add(aVal);
      }
       }
       return r;
    }
    测试代码如下
    public static void main(String[] args){ 
       List<Map<String,Object>> data = new ArrayList<>();
       Map<String,Object> map = new HashMap<>();
       map.put("id",1);
       map.put("pid",0);
       map.put("name","甘肃省");
       data.add(map);
       Map<String,Object> map2 = new HashMap<>();
       map2.put("id",2);
       map2.put("pid",1);
       map2.put("name","天水市");
       data.add(map2);
       Map<String,Object> map3 = new HashMap<>();
       map3.put("id",3);
       map3.put("pid",2);
       map3.put("name","秦州区");
       data.add(map3);
       Map<String,Object> map4 = new HashMap<>();
       map4.put("id",4);
       map4.put("pid",0);
       map4.put("name","北京市");
       data.add(map4);
       Map<String,Object> map5 = new HashMap<>();
       map5.put("id",5);
       map5.put("pid",4);
       map5.put("name","昌平区");
       data.add(map5);
       System.out.println(JSON.toJSONString(data));
       JSONArray result = listToTree(JSONArray.parseArray(JSON.toJSONString(data)),"id","pid","children");
       System.out.println(JSON.toJSONString(result));
    }