集合根据父子节点进行排序

集合根据父子节点进行排序

private List<ResAreaBean> sortParentAndChild(List<ResAreaBean> entities) {
		// 1. 寻找集合中的所有根节点
		Map<String, List<ResAreaBean>> pMap = new HashMap<String, List<ResAreaBean>>(); // 父节点为key
		Set<String> ids = new HashSet<String>(); // 存储节点id
		Set<String> pids = new HashSet<String>(); //存储父节点id
		for (ResAreaBean entity : entities) {
			ids.add(entity.getTreeCode());
			String pid = entity.getParentAreaCode();
			pids.add(pid);
			if(null == pMap.get(pid)) {
				pMap.put(pid, new ArrayList<ResAreaBean>());
			}
			pMap.get(pid).add(entity);
		}
		pids.removeAll(ids); // 得到根节点

		// [OPTIONAL] 节点id重复,退出
		if (ids.size() < entities.size()) {
			return null;
		}

		// 2. 父子排序
		List<ResAreaBean> sortedList= new ArrayList<ResAreaBean>();
		for (String rootPid : pids) {
			List<ResAreaBean> queue = pMap.remove(rootPid);
			if (null != queue) {
				while(queue.size() > 0) {
					ResAreaBean entity = queue.remove(0);
					sortedList.add(entity);
					List<ResAreaBean> tmpList = pMap.remove(entity.getTreeCode());
					if (null != tmpList) {
						queue.addAll(tmpList);
					}
				}
			}
		}
		return sortedList;
	}
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值