根据父级id查找所有的子集id（包括子集也可以当父级）

无遇696

于 2024-08-05 11:24:58 发布

阅读量225

点赞数 9

文章标签： java

本文链接：https://blog.csdn.net/weixin_68338449/article/details/140922382

版权

public static List<BDocument> findChildren(List<BDocument> documents, Long parentId) {
    Map<Long, BDocument> personMap = new HashMap<>();
    for (BDocument p : documents) {
        personMap.put(p.getId(), p);
    }

    List<BDocument> children = new ArrayList<>();
    for (BDocument p : documents) {
        if (p.getPid() != null && p.getPid().equals(parentId)) {
            children.add(p);
            findChildrenRecursively(personMap, p, children);
        }
    }
    return children;
}

private static void findChildrenRecursively(Map<Long, BDocument> documentMap, BDocument parent, List<BDocument> children) {
    Long currentParentId = parent.getId();
    for (BDocument p : documentMap.values()) {
        if (p.getPid() != null && p.getPid().equals(currentParentId)) {
            children.add(p);
            findChildrenRecursively(documentMap, p, children);
        }
    }
}