获取Manager java下的所有员工列表

I want to create a Java hierarchy problem where i want to get list of all employees working under given manager.

我想创建一个Java层次结构问题，我想获得在给定经理下工作的所有员工的列表。

My first line of input contains managerId for whom i need to get all employee list.

我的第一行输入包含managerId，我需要为其获取所有员工列表。

From second line onward i'll have managerid at index 0 and subsequent all reportees. Example:

从第二行开始，我将在指数0处有经理人，随后是所有报告人。例：

Input :

输入：

2

2

1234

1234

2567

2567

589

589

Output:

输出：

56789

56789

public class Solution {
  public static void main(String[] args) {
    Scanner scan = new Scanner(System.in);
    int toFind = scan.nextInt();
    scan.nextLine();

    while (scan.hasNext()) {
        String str = scan.nextLine();
        char[] arr = str.toCharArray();

        int managerId = Integer.parseInt(String.valueOf(arr[0]));

        for (int idx = 1; idx < arr.length; idx++) {
            Employee.addEmployee(new Employee(Integer.parseInt(String.valueOf(arr[idx])), managerId));
        }

    }

    System.out.println(Employee.getReporteeList(toFind));
}
}

class Employee {

private int employeeId;
private int managerId;
private static Set<Integer> reporteeList = new HashSet<Integer>();
public static Map<Integer, Employee> employeeMap = new HashMap<Integer, Employee>();

public Employee(int employeeId, int managerId) {
    this.employeeId = employeeId;
    this.managerId = managerId;
}

public static void addEmployee(Employee emp) {
    if (employeeMap.get(emp.managerId) == null) {
        emp.reporteeList.add(emp.employeeId);
        employeeMap.put(emp.managerId, emp);
    } else {
        employeeMap.get(emp.managerId).reporteeList.add(emp.employeeId);
    }
}

static Set getReporteeList(int toFind) {

    return employeeMap.get(toFind).reporteeList;
}

@Override
public String toString() {
    // TODO Auto-generated method stub
    return "" + employeeId;
}
}

Please help what is missing in above code so that i can get list of all employee properly.

请帮助上面代码中缺少的内容，以便我可以正确获取所有员工的列表。

1 个解决方案

#1

I'm suggesting to have a single Map to hold the data of the manager id to the list of reportees. Then given a manager id the reportee list can be obtained by looking into the same map for each of the direct reportees.

我建议使用一个Map将经理ID的数据保存到reportees列表中。然后给定经理ID，可以通过查看每个直接报告者的相同地图来获得报告者列表。

Map<Integer, List<Integer>> managerToReportees = new HashMap<>();
//Code to construct it/insert into it.

public List<Integer> getReportees(Integer employeeId) {
    List<Integer> reportees = new ArrayList<>();
    if (managerToReportees.containsKey(employeeId)) { //If the employee is a manager
       reportees.addAll(managerToReportees.get(employeeId)); //add all direct reports
       for(Integer empId : managerToReportees.get(employeeId)) { //for each of the direct reports recursively get the reportees
          reportees.addAll(getReportees(empId));
       }
    }
    return reportees; //return list of reportees for the current employeeId
}