java 获取集合中的父id

Question

一个List单位对象集合，主要字段是id parentId（本单位的父单位id，有多级） UnionCode（父code是2位数，子级每级递增2位）需要查找出这个集合中所存在的最上级的单位，（假如有父级单位A，子级单位为A1，A1的子级单位A2，集合中存在A1不存在A，那么最上级单位是A1，）以及如果存在中间有截断的情况，（如：集合中存在A，A2，需要将A2的parentId修改为A的id）

Answer 1

根据你的问题我第一反应用递归来做， 遍历集List合找到没有父单位的单位作为候选的最上级单位，然后递归地查找父级单位直到找到最顶层的单位

class Unit {
    Integer id;
    Integer parentId;
    String unionCode;
}

public Unit findTopUnit(List<Unit> units) {
    List<Unit> candidates = new ArrayList<>();
    for (Unit unit : units) {
        if (unit.parentId == null || !containsUnit(units, unit.parentId)) {
            candidates.add(unit);
        }
    }
    if (candidates.isEmpty()) {
        return null;
    }
    Unit topUnit = candidates.get(0);
    for (Unit candidate : candidates) {
        if (isAncestor(units, candidate, topUnit)) {
            topUnit = candidate;
        }
    }
    return topUnit;
}

private boolean isAncestor(List<Unit> units, Unit ancestor, Unit unit) {
    if (ancestor.id.equals(unit.id)) {
        return true;
    }
    String parentUnionCode = getParentUnionCode(unit.unionCode);
    Integer parentId = getParentId(units, parentUnionCode);
    if (parentId == null) {
        Unit parentUnit = new Unit();
        parentUnit.id = -1;
        parentUnit.parentId = null;
        parentUnit.unionCode = parentUnionCode;
        units.add(parentUnit);
        parentId = parentUnit.id;
        unit.parentId = parentId;
    }
    unit.parentId = parentId;
    return isAncestor(units, ancestor, getUnitById(units, parentId));
}

private Integer getParentId(List<Unit> units, String unionCode) {
    for (Unit unit : units) {
        if (unit.unionCode.equals(unionCode)) {
            return unit.id;
        }
    }
    return null;
}

private String getParentUnionCode(String unionCode) {
    int length = unionCode.length();
    if (length <= 2) {
        return null;
    }
    return unionCode.substring(0, length - 2);
}

private boolean containsUnit(List<Unit> units, Integer id) {
    for (Unit unit : units) {
        if (unit.id.equals(id)) {
            return true;
        }
    }
    return false;
}

private Unit getUnitById(List<Unit> units, Integer id) {
    for (Unit unit : units) {
        if (unit.id.equals(id)) {
            return unit;
        }
    }
    return null;
}