Java中使用递归算法实现查找树形结构中所有父级和子级节点
程序员文章站
2024-03-19 19:26:22
...
场景
在企业架构管理中使用树形结构进行管理,如图:
注:如果A的id是B的pid,那么A就是B的父级。
数据库数据如下:
现在需要根据传递的id查询此节点所有的父级节点以及此节点所有的子级节点。
实现
递归查询父级节点
1.一个节点只能有一个父级节点。
2.首先根据参数传递的id属性查询到pid为这个id的对象a,然后再获取这个刚查询的对象a的id,再查询pid为对象a的id的对象b,以此类推。
List<Long> result1 = new ArrayList<Long>();
public void selectFather(Long id){
//查询数据库中对应id的实体类
SysEnterpriseOrg sysEnterpriseOrg = sysEnterpriseOrgMapper.selectById(id);
//查询父级架构
//此处使用mybaatisPlus的条件构造器,查询id等于pid的对象
QueryWrapper<SysEnterpriseOrg> sysEnterpriseOrgChildQueryWrapper = new QueryWrapper<SysEnterpriseOrg>();
sysEnterpriseOrgChildQueryWrapper.eq("id",sysEnterpriseOrg.getPid());
//查询出符合条件的对象
sysEnterpriseOrg= sysEnterpriseOrgMapper.selectOne(sysEnterpriseOrgChildQueryWrapper);
if(sysEnterpriseOrg!=null){
//如果查出的对象不为空,则将此对象的id存到全局变量中,并且继续调用自己,即递归,一直到查询不到为止
result1.add(sysEnterpriseOrg.getId());
selectFather(sysEnterpriseOrg.getId());
}
}
递归查询子级节点
1.一个节点可能有多个子级节点,每个自己节点可能还有更多的子级节点。
2.所以递归时的参数用一个list来接受,首先遍历参数list,分别查询pid为参数id的对象。
3.每一个参数id所查询返回的数据是一个对象的list。
4.遍历list获取符合条件的对象的id值,一份存到temp中用作递归的参数,并存到全局变量中用来获取所有符合条件的id。
List<Long> result = new ArrayList<Long>();
public void selectChild(List<Long> ids){
//用来存取调用自身递归时的参数
List<Long> temp= new ArrayList<Long>();
//查询数据库中对应id的实体类
List<SysEnterpriseOrg> sysEnterpriseOrgList = new ArrayList<SysEnterpriseOrg>();
//遍历传递过来的参数ids
for (Long id :ids) {
//查询子级架构
//此处使用mybaatisPlus的条件构造器,查询pid等于id的对象
QueryWrapper<SysEnterpriseOrg> sysEnterpriseOrgChildQueryWrapper = new QueryWrapper<SysEnterpriseOrg>();
sysEnterpriseOrgChildQueryWrapper.eq("pid",id.toString());
//查询结果返回一个list
sysEnterpriseOrgList= sysEnterpriseOrgMapper.selectList(sysEnterpriseOrgChildQueryWrapper);
//遍历list获取符合条件的对象的id值,一份存到temp中用作递归的参数,并存到全局变量中用来获取所有符合条件的id
for (SysEnterpriseOrg s:sysEnterpriseOrgList) {
temp.add(s.getId());
result.add(s.getId());
}
}
if(temp.size()!=0&&temp!=null){
selectChild(temp);
}
}
上一篇: 自动化测试框架之selenium学习笔记
下一篇: 使用Rust递归实现斐波那契数列