Java中使用递归算法实现查找树形结构中所有父级和子级节点

短命女 2023-10-18 20:01 31阅读 0赞

## 场景 ##

在企业架构管理中使用树形结构进行管理，如图：

![Image 1][]![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70][]

注：如果A的id是B的pid，那么A就是B的父级。

数据库数据如下：

![Image 1][]![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70 1][]

现在需要根据传递的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);
            }
        }

[Image 1]: 
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20190618141748318.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk=,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20190618141751382.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0JBREFPX0xJVU1BTkdfUUlaSEk=,size_16,color_FFFFFF,t_70