java 递归操作

Java 递归 代码实现

多层分级处理

    public List<TreeTest> treeList() {
        List<TreeTest> result = new ArrayList<>();
        // 获取全部数据
        List<TreeTest> list = sysMenuMapper.queryTreeList();
        // 遍历list，获取父节点为空的数据
        for (TreeTest tree : list) {
            // 获取最顶层的一列数据
            if (tree.getPid() == null) {
                result.add(tree);
            }
        }
        // 递归设置子节点信息
        for (TreeTest tree1 : result) {
            setChildren(tree1, list);
        }
        return result;
    }
    private void setChildren(TreeTest tree1, List<TreeTest> list) {
        // 获取sysMenu节点的所有子节点
        Set<TreeTest> children = getChildren(list, tree1.getId());
        // 设置children节点信息
        tree1.setChildren(children);
        // 遍历list，将该节点的父节点添加到children集合中
        for (TreeTest tree3 : list) {
            if (tree1.getId() != null && tree1.getId() == tree3.getPid()) {
                // 循环调用符合条件的 setChildren ，不断往后叠加，递归调用
                setChildren(tree3, list);
                children.add(tree3);
            }
        }
    }
    // 根据节点id获取节点的所有子节点集合
    private Set<TreeTest> getChildren(List<TreeTest> list, int id) {
        // 空的子节点集合children
        Set<TreeTest> children = new HashSet<TreeTest>();
        for (TreeTest tree2 : list) {
            // 如果temp的pid不为空，且等于id,添加到children(下一层的pid等于上一层的id)
            if (tree2.getPid() != null && tree2.getPid() == id) {
                children.add(tree2);
            }
        }
        return children;
    }