JAVA将List转成Tree树形结构数据和深度优先遍历

引言:

在日常开发中,我们经常会遇到需要将数据库中返回的数据转成树形结构的数据返回,或者需要对转为树结构后的数据绑定层级关系再返回,比如需要统计当前节点下有多少个节点等,因此我们需要封装一个ListToTree的工具类和学会如何通过深度优先遍历数据。

数据准备:

先简单准备一下具有父子关系的数据。

package data;/**
* @author sinder
* @date 2023/11/8 21:40
*/
public class OrgData {private String id;private String pId;private String orgName;public String getId() {return id;}public void setId(String id) {this.id = id;}public String getpId() {return pId;}public void setpId(String pId) {this.pId = pId;}public String getOrgName() {return orgName;}public void setOrgName(String orgName) {this.orgName = orgName;}
}
package data;import java.util.ArrayList;
import java.util.List;/*** 生成数据,模拟数据库查询* @author sinder* @date 2023/11/8 22:17*/
public class SingData {public static List<OrgData> getData() {OrgData orgData1 = new OrgData();orgData1.setId("1");orgData1.setpId("root");orgData1.setOrgName("根节点A");OrgData orgData2 = new OrgData();orgData2.setId("2");orgData2.setpId("root");orgData2.setOrgName("根节点B");OrgData orgData3 = new OrgData();orgData3.setId("3");orgData3.setpId("1");orgData3.setOrgName("A目录");OrgData orgData4 = new OrgData();orgData4.setId("4");orgData4.setpId("2");orgData4.setOrgName("B目录");List<OrgData> list = new ArrayList<>();list.add(orgData1);list.add(orgData2);list.add(orgData3);list.add(orgData4);return list;}
}

正常情况下,我们都会选择封装一个将List转换为Tree的工具类,并且通过链式顺序LinkedList存储来遍历Tree数据,这里使用steam来实现,如下:

package data;import java.util.List;/*** 构建tree数据* @author sinder* @date 2023/11/8 22:30*/
public class TreeBean {private String treeId;private String treePid;private String orgName;private Integer flag = 0;private List<TreeBean> children;private OrgData orgData;public String getTreeId() {return treeId;}public void setTreeId(String treeId) {this.treeId = treeId;}public String getOrgName() {return orgName;}public void setOrgName(String orgName) {this.orgName = orgName;}public String getTreePid() {return treePid;}public void setTreePid(String treePid) {this.treePid = treePid;}public List<TreeBean> getChildren() {return children;}public void setChildren(List<TreeBean> children) {this.children = children;}public OrgData getOrgData() {return orgData;}public void setOrgData(OrgData orgData) {this.orgData = orgData;}public Integer getFlag() {return flag;}public void setFlag(Integer flag) {this.flag = flag;}@Overridepublic String toString() {return "TreeBean{" +"treeId='" + treeId + '\'' +", treePid='" + treePid + '\'' +", orgName='" + orgName + '\'' +", children=" + children +'}';}
}
package utils;import data.OrgData;
import data.SingData;
import data.TreeBean;import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;/*** 将List转为Tree** @author sinder* @date 2023/11/8 22:28*/
public class ListToTreeUtil {public static List<TreeBean> toTree(List<OrgData> list, String root) {if (list.isEmpty()) {return new ArrayList<>();}// 构建数据List<TreeBean> treeBeans = list.stream().map(item -> {TreeBean treeBean = new TreeBean();treeBean.setTreeId(item.getId());treeBean.setTreePid(item.getpId());treeBean.setOrgName(item.getOrgName());return treeBean;}).collect(Collectors.toList());// 构建Map数据Map<String, List<TreeBean>> treeMap = treeBeans.stream().collect(Collectors.groupingBy(item -> {if (item.getTreePid().isEmpty()) {item.setTreePid(root);}return item.getTreePid();}));// 将数据进行分组return treeBeans.stream().peek(data -> {List<TreeBean> children = treeMap.get(data.getTreeId());if (children != null && !children.isEmpty()) {data.setChildren(children);}}).filter(data -> data.getTreePid().equals(root)).collect(Collectors.toList());}
}

执行结果:

对tree数据进行遍历,通过链表的形式:

import data.OrgData;
import data.SingData;
import data.TreeBean;
import utils.ListToTreeUtil;import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;/*** @author sinder* @date 2023/11/8 20:44*/
public class Main {public static void main(String[] args) {// 模拟查询数据List<OrgData> orgDataList = SingData.getData();// 构建tree数据List<TreeBean> treeBean = ListToTreeUtil.toTree(orgDataList, "root");// 使用LinkedList实现链式队列,实现深度遍历LinkedList<TreeBean> stack = new LinkedList<>();stack.addAll(treeBean);while (!stack.isEmpty()) {// 从栈顶开始访问TreeBean pop = stack.peek();// Flag=1表示已经遍历过一次且该节点存在子节点if (pop.getFlag() == 1) {OrgData orgData =pop.getOrgData();List<TreeBean> children = pop.getChildren();// 获取子节点的节点名称,也可以进行其他的操作List<String> collect = children.stream().map(TreeBean::getOrgName).collect(Collectors.toList());StringBuilder builder = new StringBuilder();for (String s : collect) {builder.append(s);builder.append(">");}pop.setOrgName(builder.toString());orgData.setOrgName(pop.getOrgName());// pop出栈,当前节点已经统计完,出栈获取下一个栈顶peekstack.pop();} else {// flag为0表示未遍历,判断是否已经遍历到叶子节点(最底部)if (pop.getChildren()!= null && !pop.getChildren().isEmpty()) {// 非叶子节点pop.setFlag(1);List<TreeBean> children = pop.getChildren();for (TreeBean child : children) {// 将叶子节点入栈,放到栈顶,实现深度遍历,nextstack.push(child);}} else {// 叶子节点直接出栈即可,cnt为本身stack.pop();}}}// 遍历最终的数据for (OrgData orgData : orgDataList) {System.out.println(orgData.toString());}}
}

但是现在有一个问题,当我们响应的数据从OrgData换到其他类型的时候,这时候就需要封装成一个泛型类使得我们的tree数据生成类变成一个通用的,完整代码如下:

完整代码:JAVA将List转为Tree和深度优先遍历

package data;import java.util.List;/*** 定义一个接口,使得每一个响应的数据实体来实现* @author sinder* @date 2023/11/9 0:31*/
public interface Tree<T> {String getTreeId();String getTreePid();void setChild(List<T> list);
}
package data;import java.util.List;/*** 实现Tree<OrgData>并定义响应类型为本身* 定义转为树的参数返回
* @author sinder
* @date 2023/11/8 21:40
*/
public class OrgData implements Tree<OrgData>{private String id;private String pId;private String orgName;// 转成树需要的参数private Integer flag = 0;private List<OrgData> child;public String getId() {return id;}public void setId(String id) {this.id = id;}public String getpId() {return pId;}public void setpId(String pId) {this.pId = pId;}public String getOrgName() {return orgName;}public void setOrgName(String orgName) {this.orgName = orgName;}public Integer getFlag() {return flag;}public void setFlag(Integer flag) {this.flag = flag;}public List<OrgData> getChild() {return child;}@Overridepublic String toString() {return "OrgData{" +"id='" + id + '\'' +", pId='" + pId + '\'' +", orgName='" + orgName + '\'' +'}';}@Overridepublic String getTreeId() {return id;}@Overridepublic String getTreePid() {return pId;}@Overridepublic void setChild(List<OrgData> list) {this.child = list;}
}

ListToTree方法

package utils;import data.OrgData;
import data.SingData;
import data.Tree;
import data.TreeBean;import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;/*** 将List转为Tree* <T extends Tree>告诉编译器有Tree的get方法** @author sinder* @date 2023/11/8 22:28*/
public class ListToTreeUtil {public static <T extends Tree> List<T> toTree(List<T> list, String root) {// 当根节点为null时,定义一个初始值,防止nullString treeRoot = "treeRoot";String finalRoot = root;if (list.isEmpty()) {return new ArrayList<>();}// 构建Map数据// 根据pid分组,获取所有的子节点集合Map<String, List<T>> childMap =list.stream().collect(Collectors.groupingBy(item -> {String treePid = item.getTreePid();if (treePid == null) {treePid = treeRoot;}return treePid;}));return list.stream().peek(data -> {List<T> children = childMap.get(data.getTreeId());if (children != null && !children.isEmpty()) {data.setChild(children);}}).filter(data -> data.getTreePid().equals(finalRoot)).collect(Collectors.toList());}
}

深度优先遍历

import data.OrgData;
import data.SingData;
import data.TreeBean;
import utils.ListToTreeUtil;import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;/*** @author sinder* @date 2023/11/8 20:44*/
public class Main {public static void main(String[] args) {// 模拟查询数据List<OrgData> orgDataList = SingData.getData();// 构建tree数据List<OrgData> treeBean = ListToTreeUtil.toTree(orgDataList, "root");// 使用LinkedList实现链式队列,实现深度遍历LinkedList<OrgData> stack = new LinkedList<>();stack.addAll(treeBean);while (!stack.isEmpty()) {// 从栈顶开始访问OrgData pop = stack.peek();// Flag=1表示已经遍历过一次且该节点存在子节点if (pop.getFlag() == 1) {List<OrgData> children = pop.getChild();// 获取子节点的节点名称,也可以进行其他的操作List<String> collect = children.stream().map(OrgData::getOrgName).collect(Collectors.toList());StringBuilder builder = new StringBuilder();for (String s : collect) {builder.append(s);builder.append(">");}pop.setOrgName(builder.toString());// pop出栈,当前节点已经统计完,出栈获取下一个栈顶peekstack.pop();} else {// flag为0表示未遍历,判断是否已经遍历到叶子节点(最底部)if (pop.getChild()!= null && !pop.getChild().isEmpty()) {// 非叶子节点pop.setFlag(1);List<OrgData> children = pop.getChild();for (OrgData child : children) {// 将叶子节点入栈,放到栈顶,实现深度遍历,nextstack.push(child);}} else {// 叶子节点直接出栈即可,cnt为本身stack.pop();}}}// 遍历最终的数据for (OrgData orgData : orgDataList) {System.out.println(orgData.toString());}}
}

        文章主要讲了tree数据的生成策略和对tree数据的深度遍历,整体上先构建出一个tree数据,为了能兼容多Object数据转为tree之后的遍历,因此使用了泛型,并且定义了Tree接口,提供了getid、getPid、setChild等操作方法,实体类在实现了Tree接口后就可以使用相应的方法来操作父子关系相关的ip和pid以及子节点来构建相关的tree数据,这也方便的在使用LinkedList深度遍历tree能更加灵活的操作父子节点的数据,主要通过出栈入栈来实现深度遍历,一路从父节点一直遍历到叶子节点才停止。

文章仅为参考示例,更多更好的实现方式欢迎留言评论呀!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/135591.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Python 海龟绘图基础教学教案(二十六)

Python 海龟绘图——第 49 题 题目&#xff1a;绘制下面的图形 解析&#xff1a; 使用二重循环绘制六叶长方形风车。 答案&#xff1a; Python 海龟绘图——第 50 题 题目&#xff1a;绘制下面的图形 解析&#xff1a;使用二重循环绘制由四个相同大小菱形组成的四叶风车图案…

朴素贝叶斯算法

朴素贝叶斯(naive Bayes)算法是基于贝叶斯定理与特征条件独立假设的分类方法。对于给定的训练数据集&#xff0c;首先基于特征条件独立假设学习输入输出的联合概率分布。然后基于此模型&#xff0c;对给定的输入x&#xff0c;利用贝叶斯定理求出后验概率最大的输出y。不同于其他…

opengauss权限需求

创建角色 "u_rts" 并授予对数据库 "rts_opsdb" 的只读权限&#xff1a; CREATE ROLE u_rts LOGIN PASSWORD Cloud1234; GRANT CONNECT ON DATABASE rts_opsdb TO u_rts; GRANT USAGE ON SCHEMA public TO u_rts; GRANT SELECT ON ALL TABLES IN SCHEMA pub…

STM32MPU6050角度的读取(STM32驱动MPU6050)

注&#xff1a;文末附STM32驱动MPU6050代码工程链接&#xff0c;需要的读者请自取。 一、MPU6050介绍 MPU6050是一款集成了三轴陀螺仪和三轴加速度计的传感器芯片&#xff0c;由英国飞利浦半导体&#xff08;现为恩智浦半导体&#xff09;公司生产。它通过电子接口&#xff08…

【JS】Chapter10-Bom 操作

站在巨人的肩膀上 黑马程序员前端JavaScript入门到精通全套视频教程&#xff0c;javascript核心进阶ES6语法、API、js高级等基础知识和实战教程 &#xff08;十&#xff09;Bom 操作 1. Window对象 1.1 BOM(浏览器对象模型) BOM(Browser Object Model) 是浏览器对象模型wind…

力扣labuladong一刷day2共7题

力扣labuladong一刷day2共8题 | 26. 删除有序数组中的重复项 83. 删除排序链表中的重复元素 文章目录 力扣labuladong一刷day2共8题 | 26. 删除有序数组中的重复项 83. 删除排序链表中的重复元素一、26. 删除有序数组中的重复项二、83. 删除排序链表中的重复元素三、27. 移除元…

C++实战学习:输出类的抽象和实现详解

最近写了一些博客复习C的知识&#xff0c;但理论终究是理论&#xff0c;那多态、继承等C特性到底该在什么情况下使用&#xff1f;如何模块化地完成一个程序呢&#xff1f;还有没有什么C语法方面值得学习的知识呢&#xff1f;本节就来分析一个实际项目中的例子&#xff0c;来理解…

多测师肖sir_高级金牌讲师_jenkins搭建

jenkins操作手册 一、jenkins介绍 1、持续集成&#xff08;CI&#xff09; Continuous integration 持续集成 团队开发成员每天都有集成他们的工作&#xff0c;通过每个成员每天至少集成一次&#xff0c;也就意味着一天有可 能多次集成。在工作中我们引入持续集成&#xff0c;通…

大模型时代,开发者成长指南 | 新程序员

【编者按】GPT 系列的面世影响了全世界、各个行业&#xff0c;对于开发者们的感受则最为深切。以 ChatGPT、Github Copilot 为首&#xff0c;各类 AI 编程助手层出不穷。编程范式正在发生前所未有的变化&#xff0c;从汇编到 Java 等高级语言&#xff0c;再到今天以自然语言为特…

Python高级语法----Python多线程与多进程

文章目录 多线程多进程注意事项多线程与多进程是提高程序性能的两种常见方法。在深入代码之前,让我们先用一个简单的比喻来理解它们。 想象你在一家餐厅里工作。如果你是一个服务员,同时负责多个桌子的顾客,这就类似于“多线程”——同一个人(程序)同时进行多项任务(线程…

HttpUtils工具类

作为Java开发程序员&#xff0c;需要我们经常写一些工具类来简化开发过程&#xff0c;我们自己肯定写过或者用过HttpUtils用来发送http请求&#xff0c;但是每次手写太繁琐了&#xff0c;于是就按照标准写了一个Http工具类&#xff0c;现在分享出来。 1.HTTP请求简介 HTTP(Hy…

SSM 线上知识竞赛系统-计算机毕设 附源码 27170

SSM线上知识竞赛系统 摘 要 科技进步的飞速发展引起人们日常生活的巨大变化&#xff0c;电子信息技术的飞速发展使得电子信息技术的各个领域的应用水平得到普及和应用。信息时代的到来已成为不可阻挡的时尚潮流&#xff0c;人类发展的历史正进入一个新时代。在现实运用中&#…

虚幻引擎:如何使用 独立进程模式进行模拟

第一步:先更改配置 第二步,在启动的两个玩家里面,一个设为服务器,一个链接进去地图就可以了 1.设置服务器 2.另一个玩家链接

企业级低代码开发,科技赋能让企业具备“驾驭软件的能力”

科技作为第一生产力&#xff0c;其强大的影响力在各个领域中都有所体现。数字技术&#xff0c;作为科技领域中的一股重要力量&#xff0c;正在对传统的商业模式进行深度的变革&#xff0c;为各行业注入新的生命力。随着数字技术的不断发展和应用&#xff0c;企业数字化转型的趋…

远程运维用什么软件?可以保障更安全?

远程运维顾名思义就是通过远程的方式IT设备等运行、维护。远程运维适用场景包含因疫情居家办公&#xff0c;包含放假期间出现运维故障远程解决&#xff0c;包含项目太远需要远程操作等等。但远程运维过程存在一定风险&#xff0c;安全性无法保障&#xff0c;所以一定要选择靠谱…

RocketMQ如何安全的批量发送消息❓

优点&#xff1a; 批量发送消息可以提高rocketmq的生产者性能和吞吐量。 使用场景: 发送大量小型消息时&#xff1b;需要降低消息发送延迟时&#xff1b;需要提高生产者性能时&#xff1b; 注意事项&#xff1a; 消息列表的大小不能超过broker设置的最大消息大小;消息列表…

如何快速教你看自己电脑cpu是几核几线程

目录 一、我们日常中说的电脑多少核多少线程&#xff0c;很多人具体不知道什么意思&#xff0c;下面举例4核和4线程什么意思。二、那么4线程又是怎么回事呢&#xff1f;三、那么知道了上面的介绍后怎么看一台电脑是几核&#xff0c;几线程呢&#xff1f; 一、我们日常中说的电脑…

JSONP 跨域访问(2), JSONP劫持

JSONP 跨域访问(2), JSONP劫持 一, 利用 XSS 漏洞执行jsonp 1. 利用过程 发现有jsonp的请求: <script type"text/javascript" src"http://192.168.112.200/security/jsonp.php?callbackjsonpCallback"></script>向xss漏洞的位置注入代码…

​软考-高级-信息系统项目管理师教程 第四版【第24章-法律法规与标准规范-思维导图】​

软考-高级-信息系统项目管理师教程 第四版【第24章-法律法规与标准规范-思维导图】 课本里章节里所有蓝色字体的思维导图

springboot 项目升级 2.7.16 踩坑

记录一下项目更新版本依赖踩坑 这个是项目最早的版本依赖 这里最初是最初是升级到 2.5.7 偷了个懒 这个版本的兼容性比较强 就选了这版本 也不用去修改就手动的去换了一下RabbitMQ的依赖 因为这边项目有AMQP 风险预警 1.spring-amqp版本低于2.4.17的用户应升级到2.4.17 2.spri…