注册 登录
编程论坛 JAVA论坛

Java多叉树递归算法构造树形结构JSON

人知了 发布于 2022-03-26 11:08, 2377 次点击
构造如下JSON数据(很多树形组件都需要树形结构JSON):
程序代码:

{
  id: '100000',
  text: '廊坊银行总行',
  children: [
    {
      id: '110000',
      text: '廊坊分行',
      children: [
        {
          id: '113000',
          text: '廊坊银行开发区支行',
          leaf: true
        },
        {
          id: '112000',
          text: '廊坊银行解放道支行',
          children: [
            {
              id: '112200',
              text: '廊坊银行三大街支行',
              leaf: true
            },
            {
              id: '112100',
              text: '廊坊银行广阳道支行',
              leaf: true
            }
          ]
        },
        {
          id: '111000',
          text: '廊坊银行金光道支行',
          leaf: true
        }
      ]
    }
  ]
}




实现这样一颗树,需要设计两个类:树类(MultiwayTree)、节点类(Node);排序时还需要一个比较器类(NodeIDComparator);为了方便演示,还需要构造一些假的层次数据,因此还需要建一个构造假数据的类(VirtualDataGenerator),以下代码拷贝出来之后可直接运行测试:
程序代码:

package test;

import java.util.ArrayList;
import java.import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Collections;

/**

 * 多叉树类

 
*/
public class MultiwayTree {
    public static void main(String[] args) {
        // 读取层次数据结果集列表
        List dataList = VirtualDataGenerator.getVirtualResult();

        // 节点映射表,用于临时存储节点对象
        HashMap nodeMap = new HashMap();
        // 根节点
        Node root = null;
        // 将结果集存入映射表(后面将借助映射表构造多叉树)
        for (Iterator it = dataList.iterator(); it.hasNext();) {
            Map dataRecord = (Map) it.next();
            Node node = new Node();
            node.id = (String) dataRecord.get("id");
            node.text = (String) dataRecord.get("text");
            node.parentId = (String) dataRecord.get("parentId");
            nodeMap.put(node.id, node);
        }
        // 构造无序的多叉树
        Set entrySet = nodeMap.entrySet();
        for (Iterator it = entrySet.iterator(); it.hasNext();) {
            Node node = (Node) ((Map.Entry) it.next()).getValue();
            if (node.parentId == null || node.parentId.equals("")) {
                root = node;
            } else {
                ((Node) nodeMap.get(node.parentId)).addChild(node);
            }
        }
        // 输出无序的树形结构的JSON字符串
        System.out.println(root);
        // 对多叉树进行横向排序
        root.sortChildren();
        // 输出有序的树形结构的JSON字符串
        System.out.println(root);

        // 程序输出结果如下:
        
//
        
// 无序的树形结构(格式化后的结果,可使用JSON格式化工具查看,例如  http://jsonviewer.stack.hu/ 在线查看器):  
        
//  {
        
//   id : '100000',
        
//   text : '廊坊银行总行',
        
//   children : [
        
//     {
        
//     id : '110000',
        
//     text : '廊坊分行',
        
//     children : [
        
//       {
        
//       id : '113000',
        
//       text : '廊坊银行开发区支行',
        
//       leaf : true
        
//       },
        
//       {
        
//       id : '111000',
        
//       text : '廊坊银行金光道支行',
        
//       leaf : true
        
//       },
        
//       {
        
//       id : '112000',
        
//       text : '廊坊银行解放道支行',
        
//       children : [
        
//         {
        
//         id : '112200',
        
//         text : '廊坊银行三大街支行',
        
//         leaf : true
        
//         },
        
//         {
        
//         id : '112100',
        
//         text : '廊坊银行广阳道支行',
        
//         leaf : true
        
//         }
        
//       ]
        
//       }
        
//     ]
        
//     }
        
//   ]
        
//  }

        
// 有序的树形结构(格式化后的结果):
        
//  {
        
//   id : '100000',
        
//   text : '廊坊银行总行',
        
//   children : [
        
//     {
        
//     id : '110000',
        
//     text : '廊坊分行',
        
//     children : [
        
//       {
        
//       id : '111000',
        
//       text : '廊坊银行金光道支行',
        
//       leaf : true
        
//       },
        
//       {
        
//       id : '112000',
        
//       text : '廊坊银行解放道支行',
        
//       children : [
        
//         {
        
//         id : '112100',
        
//         text : '廊坊银行广阳道支行',
        
//         leaf : true
        
//         },
        
//         {
        
//         id : '112200',
        
//         text : '廊坊银行三大街支行',
        
//         leaf : true
        
//         }
        
//       ]
        
//       },
        
//       {
        
//       id : '113000',
        
//       text : '廊坊银行开发区支行',
        
//       leaf : true
        
//       }
        
//     ]
        
//     }
        
//   ]
        
//  }  

    }

}

/**

 * 节点类

 
*/
class Node {
    /**
     * 节点编号
     
*/
    public String id;

    /**
     * 节点内容
     
*/
    public String text;

    /**
     * 父节点编号
     
*/
    public String parentId;

    /**
     * 孩子节点列表
     
*/
    private List children = new ArrayList();

    // 添加孩子节点
    public void addChild(Node node) {
        children.add(node);
    }

    // 先序遍历,拼接JSON字符串
    public String toString() {
        String result = "{" + "id : '" + id + "'" + ", text : '" + text + "'";
        if (children.size() != 0) {
            result += ", children : [";
            for (int i = 0; i < children.size(); i++) {
                result += ((Node) children.get(i)).toString() + ",";            
            }
            result = result.substring(0, result.length() - 1);
            result += "]";
        } else {
            result += ", leaf : true";
        }
        return result + "}";
    }

    // 兄弟节点横向排序
    public void sortChildren() {
        if (children.size() != 0) {
            // 对本层节点进行排序(可根据不同的排序属性,传入不同的比较器,这里 传入ID比较器)
            Collections.sort(children, new NodeIDComparator());
            // 对每个节点的下一层节点进行排序            
            for (int i = 0; i < children.size(); i++) {
                ((Node) children.get(i)).sortChildren();
            }
        }
    }

}

/**

 * 节点比较器

 
*/
class NodeIDComparator implements Comparator {
    // 按照节点编号比较
    public int compare(Object o1, Object o2) {
        int j1 = Integer.parseInt(((Node) o1).id);
        int j2 = Integer.parseInt(((Node) o2).id);
        return (j1 < j2 ? -1 : (j1 == j2 ? 0 : 1));
    }
}

/**

 * 构造虚拟的层次数据

 
*/
class VirtualDataGenerator {
    // 构造无序的结果集列表,实际应用中,该数据应该从数据库中查询获得;
    public static List getVirtualResult() {
        List dataList = new ArrayList();

        HashMap dataRecord1 = new HashMap();
        dataRecord1.put("id", "112000");
        dataRecord1.put("text", "廊坊银行解放道支行");
        dataRecord1.put("parentId", "110000");

        HashMap dataRecord2 = new HashMap();
        dataRecord2.put("id", "112200");
        dataRecord2.put("text", "廊坊银行三大街支行");
        dataRecord2.put("parentId", "112000");

        HashMap dataRecord3 = new HashMap();
        dataRecord3.put("id", "112100");
        dataRecord3.put("text", "廊坊银行广阳道支行");
        dataRecord3.put("parentId", "112000");

        HashMap dataRecord4 = new HashMap();
        dataRecord4.put("id", "113000");
        dataRecord4.put("text", "廊坊银行开发区支行");
        dataRecord4.put("parentId", "110000");

        HashMap dataRecord5 = new HashMap();
        dataRecord5.put("id", "100000");
        dataRecord5.put("text", "廊坊银行总行");
        dataRecord5.put("parentId", "");

        HashMap dataRecord6 = new HashMap();
        dataRecord6.put("id", "110000");
        dataRecord6.put("text", "廊坊分行");
        dataRecord6.put("parentId", "100000");

        HashMap dataRecord7 = new HashMap();
        dataRecord7.put("id", "111000");
        dataRecord7.put("text", "廊坊银行金光道支行");
        dataRecord7.put("parentId", "110000");

        dataList.add(dataRecord1);
        dataList.add(dataRecord2);
        dataList.add(dataRecord3);
        dataList.add(dataRecord4);
        dataList.add(dataRecord5);
        dataList.add(dataRecord6);
        dataList.add(dataRecord7);

        return dataList;
    }
}
0 回复
1