递归树结构

查找全部节点:

查找具体节点的子节点,孙子节点:


上面的图是用java递归树结构运行的效果,和预测的结果相同。接下来是代码实现:

import java.util.*;
public class LookWord {

	private ArrayList<TreeBean> treeList = new ArrayList<TreeBean>();  //初始

	public static void main(String[] args) {
		LookWord lw = new LookWord();
		lw.init(); // 初始化数据
		//lw.findAll(lw.treeList);  //查找指定的集合
        lw.findWho("11",lw.treeList); //找到指定的id 的集合
	}

	 // 查找id的父节点
	public void findWho(String id,ArrayList<TreeBean> ts) {
		for (TreeBean t : ts) {
			if(t.getId().equals(id)&&t.getChildren()!=null){  //找到了那个对象就不断查找出来
				  findAll(t.getChildren());
			}else{
				if(t.getChildren()!=null){
					findWho(id,t.getChildren());
				}
			}
		}
	}

	// 根据查找的集合查找
	public void findAll(ArrayList<TreeBean> findList) {
		// 根据集合的大小找子集合作为查找集合的目标
		for (int i = 0; i < findList.size(); i++) {
			TreeBean tb = findList.get(i);
			System.out.println(tb.toString()); // 不管有没有子节点都要进行输出
			if (tb.getChildren() != null) { // 如果有子节点则继续输出
				findAll(tb.getChildren()); // 如果子节点不为空则继续查找
			}
		}
	}
	

	// 初始化
	private void init() {
		ArrayList<TreeBean> children111 = new ArrayList<TreeBean>();
		TreeBean tb111 = new TreeBean("1111", "one1111", "111");
		TreeBean tb112 = new TreeBean("1112", "one1112", "111");
		children111.add(tb111);
		children111.add(tb112);
		TreeBean tb11 = new TreeBean("111", "one111", "11");
		tb11.setChildren(children111);
		TreeBean tb12 = new TreeBean("112", "one112", "11");
		TreeBean tb13 = new TreeBean("113", "one113", "11");
		ArrayList<TreeBean> children11 = new ArrayList<TreeBean>();
		children11.add(tb11);
		children11.add(tb12);
		children11.add(tb13);
		// 为11 设置子节点
		TreeBean tp11 = new TreeBean("11", "one1", "1");
		tp11.setChildren(children11);
		TreeBean tp12 = new TreeBean("12", "one2", "1");
		ArrayList<TreeBean> children1 = new ArrayList<TreeBean>();
		children1.add(tp11);
		children1.add(tp12);
		// 为1设置12,11子节点
		TreeBean tb1 = new TreeBean("1", "one", "0");
		tb1.setChildren(children1);
		ArrayList<TreeBean> children2 = new ArrayList<TreeBean>();
		children2.add(new TreeBean("21", "two1", "2"));
		children2.add(new TreeBean("22", "two2", "2"));
		// 为2设置子节点
		TreeBean tb2 = new TreeBean("2", "two", "0");
		tb2.setChildren(children2);
		treeList.add(tb1);
		treeList.add(tb2);
	}
}



/** * 根据等级查询类目 * * @param level * @return */ @Override public List queryCategoryTree(Integer level) { //查询当前级别下类目 List list = categoryDAO.list(level); //组装好的类目,返回前端 List categoryTree = new ArrayList(); //所有类目 List allDTOList = new ArrayList(); if (CollectionUtils.isEmpty(list)) { return categoryTree; } for (CategoryDO categoryDO : list) { allDTOList.add(new CategoryTreeDTO().convertDOToDTO(categoryDO)); } //当前等级类目 categoryTree = allDTOList.stream().filter(dto -> level.equals(dto.getLevel())).collect(Collectors.toList()); for (CategoryTreeDTO categoryTreeDTO : categoryTree) { //组装类目为树结构 assembleTree(categoryTreeDTO, allDTOList,Constants.CATEGORY_MAX_LEVEL - level); } return categoryTree; } /** * 组装 * * @param categoryTreeDTO * @param allList * @param remainRecursionCount 剩余递归次数 * @return */ public CategoryTreeDTO assembleTree(CategoryTreeDTO categoryTreeDTO, List allList, int remainRecursionCount) { remainRecursionCount--; //最大递归次数不超过Constants.CATEGORY_MAX_LEVEL-level次,防止坏数据死循环 if(remainRecursionCount < 0){ return categoryTreeDTO; } String categoryCode = categoryTreeDTO.getCategoryCode(); Integer level = categoryTreeDTO.getLevel(); //到达最后等级返回 if (Constants.CATEGORY_MAX_LEVEL == level) { return categoryTreeDTO; } //子类目 List child = allList.stream().filter(a -> categoryCode.equals(a.getParentCode())).collect(Collectors.toList()); if (null == child) { return categoryTreeDTO; } categoryTreeDTO.setChildren(child); //组装子类目 for (CategoryTreeDTO dto : child) { assembleTree(dto, allList,remainRecursionCount); } return categoryTreeDTO; }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值