【优雅的递归】

LFW福哥

已于 2022-08-18 18:58:44 修改

阅读量156

点赞数

分类专栏： Java知识总结文章标签： java

于 2022-08-18 18:55:20 首次发布

本文链接：https://blog.csdn.net/lfwlfwl/article/details/126411706

版权

Java知识总结专栏收录该内容

21 篇文章 0 订阅

订阅专栏

Java8优雅的递归

/**
	 * 封装树形结构
	 * @param data
	 * @return
	 */
	public List<TreeVo> listWithTree(List<TreeVo> data) {
	    //将数据组成树形结构
	    //筛选出根元素
	    //利用stream函数来过滤和处理
	    List<TreeVo> root = data.stream().filter(tmp ->
	        tmp.getParentId() == 0//获取所有父节点
	    ).map(tmps -> {
	        //设置子节点
	        tmps.setChildren(getChildrens(tmps, data));
	        return tmps;
	    }).collect(Collectors.toList()); 
	    return root;
	}

	/**
	 * 查找所有root的子节点，方法内是递归查找
	 * @param root
	 * @param all
	 * @return
	 */
	private List<TreeVo> getChildrens(TreeVo root, List<TreeVo> all) {
	    List<TreeVo> children = all.stream().filter(tmp ->
	        tmp.getParentId().equals(root.getId())
	    ).map(tmps -> {
	        //查找所有子节点的子节点
	    	tmps.setChildren(getChildrens(tmps, all));
	        return tmps;
	    }).collect(Collectors.toList());
	    return children;
	}