java 生成树形结构（极速版）

树形结构是非常常见的数据结构，生成树形结构的方法比较常见的有递归,for循环，不管使用什么方法，如果能很好地利用内存来合理地添加树节点，则能极大提高效率

测试代码如下：

public class TestTree {
	
	public static void main(String[] args) {
		C3p0Jdbc jdbc = new C3p0Jdbc();
		String sql = "SELECT code value,up_code upCode,name label,level FROM test_table WHERE level < 4 ORDER BY level";
		JSONArray arr = jdbc.query(sql);
		// 先将数据循环放入新建的map中
		Map treeMap = new HashMap<>();
		int length = arr.size();
		long t1 = System.currentTimeMillis();
		for (int i = 0; i < length; i++) {
			String key = arr.getJSONObject(i).getString("value");
			treeMap.put(key, arr.getJSONObject(i));
		}
		long t2 = System.currentTimeMillis();
		System.out.println("循环用时：" + (t2 - t1) + "ms");
		JSONArray new_tree = new JSONArray();
		// 再使用双层for循环，根据判断条件，将符合条件的节点数据添加到树形结构中
		for (int i = 0; i < length; i++) {
			JSONObject parent = arr.getJSONObject(i);
			String key = parent.getString("value");
			int level = parent.getIntValue("level");
			// 这是唯一一处需要 new 对象的地方
			JSONArray children = new JSONArray();
			for (int j = 0; j < length; j++) {
				JSONObject child = arr.getJSONObject(j);
				String childKey = child.getString("value");
				String upCode = child.getString("upCode");
				if (upCode.equals(key)) {
					children.add(treeMap.get(childKey));
				}
			}
			parent.put("children", children);
			if (level == 1) {
				new_tree.add(parent);
			}
		}
		
		long t3 = System.currentTimeMillis();
		System.out.println("第二次遍历用时：" + (t3 - t2) + "ms");
		System.out.println("全过程用时：" + (t3 - t1) + "ms");
	}
	
}

在for循环中务必要减少new 对象的次数，New 对象是很耗时的操作，不仅仅是 new 对象会很耗时，其它操作比如将一个 object 转为 jsonarray 或 json 也会很耗时，因为这种操作修改了内存中对象的引用地址，测试代码中最优势的地方就是将所有的对象都加载到map中，在new_tree.add(parent)的操作时是直接将parent对象的内存引用地址添加到了new_tree中，所以效率非常快，我的测试数据是3227条国内的省市区县数据，全过程用时在600~850 ms之间 效果如图：

感谢同事TX提供思路

转载于:https://my.oschina.net/u/4118973/blog/3086432