(礼拜二log)java 构建层级关系的递归算法

待我称王封你为后i 2022-07-24 07:14 183阅读 0赞

问题：给了一串JSON格式的数据，它们的orgId中保存着自身的ID，同时，它们还保存着父级元素的ID：orgParentId，根据这些信息，将它们按照父级、子级元素的顺序依次排列下来，如下：  
元素1:id 1，parentId：0（元素1是最高的父级元素）  
元素2:id 2，parentId：1（元素2是元素1的子元素）  
元素3:id 3，parentId：2（元素3是元素2的子元素）

元素4:id 4，parentId：0（元素4是最高的父级元素）  
…  
代码如下：  
注：page.rows中包含着原始数据

Page page = orgservice.findPage(getParames());
            JSONArray items_p = new JSONArray();
            JSONArray items_son = new JSONArray();
            JSONArray items = new JSONArray();
            for(Object obj : page.getRows()){
                JSONObject j_obj = (JSONObject)obj;
                if(j_obj.get("parentOrganizationId").equals("0") || j_obj.get("parentOrganizationId").equals("null")){
                    j_obj.put("css_id", j_obj.get("organizationId"));
                    j_obj.put("css_pid", "0");
                    items_p.add(j_obj);
                }else{
                    j_obj.put("css_id", j_obj.get("organizationId"));
                    j_obj.put("css_pid", j_obj.get("parentOrganizationId"));
                    items_son.add(j_obj);
                }
            }
            items = formatData(items_p,items_son);
            page.setRows(items);

递归函数：

// 递归方法
        protected JSONArray formatData(JSONArray p_items,JSONArray s_items){
            JSONArray items = new JSONArray();
            if(p_items.size()==0||s_items.size()==0){
                return null;
            }
            for(Object p_item : p_items){
                JSONObject J_p_item = (JSONObject)p_item;
                items.add(J_p_item);
                for(Object s_item : s_items){
                    JSONObject J_s_item = (JSONObject)s_item;
                    if(J_s_item.get("css_pid").equals(J_p_item.get("css_id"))){
                        JSONArray temp = new JSONArray();
                        temp.add(J_s_item);
                        JSONArray new_items = formatData(temp,s_items);
                        if(new_items!=null){
                            for(Object i : new_items){
                                items.add(i);
                            }
                        }else{
                            items.add(J_s_item);
                        }
                    }
                }
            }
            return items;
        }