根据ParentId生成树状结构这po事
作者:互联网
最近项目里老是碰到数据库里存放父子级关系,然后获取出来要根据ParentId生成树状结构对象的需要。
也不知道最近是脑子不行还是咋滴。这po事想了老半天才想明白怎么去构建该结构。555555最后敲出来后,豁然开朗,都有想哭的冲动了。
废话不多说,开始总结和分享一下我的解题思路。(其实这问题不难,我是菜鸡)
需求
-
数据库里有一张结构大致如下的表格:
-
我们需要冲数据库中获取出数据,然后解析成大致如下的树状格式:
解题思路
-
让我们先像个东西类比一下好吧,老习惯了,万事万物源于生活,从生活中找例子,往往更容易让人明白。
这个问题其实就相当于:
有一个家族,告诉你每个人的爸爸是谁,然后让你把他们家族谱做出来。
不就是这么一回事嘛!!!!! -
那么让我们想想族谱的问题。我们应该怎么去完成这个事情呢?我觉得首先我们可以把所有人的名单都先排成一排。如下图:
-
然后去看提供给我们的信息。我们能知道每个人的爸爸是谁。那么爸爸是谁就加谁后面。如下图:
-
最后我们会得到下图这样的结果。这个时候我们知道了什么呢?仔细观察我们就发现我们知道了,每个爸爸他有哪几个儿子。从每个人的爸爸是谁转换成了每个人有几个儿子。这一步很关键很关键。我一直没想清楚怎么解决该问题就是卡在这个点上。
-
那么为什么要知道每个人的儿子是谁呢?因为下一步我们要这么来。首先我们把最年长的对象1过来先,然后让他把自己儿子叫过来,然后儿子对象2把他的儿子叫过来。最终得到下图。
代码实现
上述废话一堆,但其实只要理解了上面的逻辑。代码就好写啦。
- 首先创建一个User类,匹配数据库的数据结构来获取数据。
@Data
@AllArgsConstructor
@NoArgsConstructor
public class User{
private Integer id;
private String name;
private Integer parentId;
}
- 首先创建一个UserVO类,作为输出的最终数据结构。
@Data
@AllArgsConstructor
@NoArgsConstructor
public class UserVO{
private Integer id;
private String name;
private List<UserVO> children;
}
- 首先通过数据库获取到所有的User
...
List<User> userList = UserMapper.getAllUser();
...
- 创建一个HashMap,遍历上一步的集合将User转变成UserVO,并且向HashMap中赋值。key是parentId,value是子类UserVO对象。还有就是创建一个UserVO集合来存放转变好的UserVO。毕竟转变好之后就不用看User了。
...
List<User> userList = UserMapper.getAllUser();
Map<Integer,List<UserVO>> parentIdUserVOMap = new HashMap<>();
List<UserVO> userVOList = new ArrayList<>();
for(User user : userList){
UserVO userVO = new UserVO();
BeanUtils.copyProperities(user,userVO);
userVOList.add(userVO);
int parentId = user.getParentId();
if(parentIdUserVOMap.containsKey(parentId)){
List<UserVO> userVOs = parentIdUserVOMap.get(parentId);
userVOs.add(userVO);
}else{
List<UserVO> userVOs = new ArrayList<>();
userVOs.add(userVO);
parentIdUserVOMap.put(parentId,userVOs);
}
}
...
- 这个时候,就相当于我们知道了所有的爸爸以及他们各自的儿子们。但是儿子们还没跟到爸爸身边。即目前每个UserVO的children属性都是空的。我们需要把子类放进去。怎么做呢?再遍历一遍所有的爸爸,让他们带好自己的儿子。
...
for(UserVO userVO : userVOList){
List<UserVO> childs = parentIdUserVOMap.get(userVO.getId());
userVO.setChildren(childs);
}
...
- 最后一步,我们要获取年纪最大的爸爸,作为树状结构的顶端,即这个人没有爸爸记录在案,parentId==0
...
List<UserVO> result = parentIdUserVOMap.get(0);
...
总结
好了,这个po问题解决了。没想通之前我是真绕了半天,也想不通为什么数据库要那么设计表。相通了之后一切都变得似乎很合理。不知道你看懂了吗?要是觉得写的不错,记得一键三连加关注哦!如果你有更好的设计思路或解决办法也可以下方留言。我就是个菜狗,还望大佬们带带我!!!
标签:...,UserVO,userVO,树状,parentId,List,private,ParentId,po 来源: https://blog.csdn.net/qq_43291207/article/details/121970048