平铺数据结构转成树状数据（双循环版）

最新推荐文章于 2023-06-04 17:18:42 发布

机械提桶跑路ing

最新推荐文章于 2023-06-04 17:18:42 发布

阅读量325

点赞数

文章标签：数据结构 java 开发语言

本文链接：https://blog.csdn.net/weixin_46703961/article/details/128908212

版权

平铺数据结构转成树状数据的方式有很多，本文采用，双循环实现，用1.8的新特性流会比较优雅，也很容易懂。

package com.ruoyi.web.controller.common;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;
import java.util.stream.Collectors;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;

/**
 * 〈〉<br>
 *
 * @author 
 * @Description
 * @date 2023/1/10 21:05
 */
@Data
@Getter
@Setter
public class test {

	int id;
	int pid;
	List<test> Children;

	/**
	 * java1.8 非递归实现平铺数据转成Tree型结构  类似于双循环实现。是吧地址塞进去了
	 */
	static final Function<List<test>,List<test>> MULTI_TREE_CONVERTER = sources->
			sources.stream()
					.filter(item->{
						item.setChildren(
								sources.stream()
										.filter(e-> Objects.equals(e.getPid(), item.getId()))
										.collect(Collectors.toList()));
						return item.getPid() == 0;})
					.collect(Collectors.toList());
	public static void main(String[] args) {
	List<test> list =new ArrayList<test>();
		test test = new test();
		test.setId(1);
		test.setPid(0);
		test t = new test();
		t.setId(2);
		t.setPid(1);
		test t1 = new test();
		t1.setId(3);
		t1.setPid(1);
		test t2 = new test();
		t2.setId(4);
		t2.setPid(2);
		list.add(test);
		list.add(t);
		list.add(t1);
		list.add(t2);
		List<com.ruoyi.web.controller.common.test> tests = buildTreeBy2Loop(list, 0);
		System.out.println(tests);
		System.out.println("-------------");
		for (com.ruoyi.web.controller.common.test test1 : MULTI_TREE_CONVERTER.apply(list)) {
			System.out.println(test1);
		}



	}
	/**
	 * 利用双循环方式构造树  
	 * @param treeNodes 所有树节点
	 * @param root 树的根节点引用
	 * @return List<BaseTreeNode> 返回构造好的树
	 */
	public static List<test> buildTreeBy2Loop(List<test> treeNodes, int root) {
		List<test> trees = new ArrayList<>();
		//判断当前节点是否是root的子节点,如果是就存入list
		for (test node : treeNodes) {
			if (root==(node.getPid())) {
				trees.add(node);
			}
			//再判断当前节点是否是其他节点的父节点,如果是就把子节点保存
			for (test treeNode : treeNodes) {
				if (node.getId()==treeNode.getPid()) {
					if (node.getChildren()==null) {
						node.setChildren(new ArrayList<>());
					}
					node.getChildren().add(treeNode);
				}
			}
		}
		return trees;
	}
}

代码完美运行，本文阐述双循环实现。