员工的重要性

/*
 * 题目思路：
 * 1. 定义一个 Employee 类，包含 id、重要度和下属 id 列表。
 * 2. 使用 HashMap 存储所有员工信息，以便快速查找员工。
 * 3. 使用 Stream 计算指定员工及其下属的重要度之和。
 */
 
import java.util.*;
import java.util.stream.*;
 
class Employee {
    public int id;
    public int importance;
    public List<Integer> subordinates;
}
 
public class Solution {
    private Map<Integer, Employee> employeeMap = new HashMap<>();
 
    public int getImportance(List<Employee> employees, int id) {
        employees.forEach(e -> employeeMap.put(e.id, e));
        return getTotalImportance(id);
    }
 
    private int getTotalImportance(int id) {
        Employee employee = employeeMap.get(id);
        return employee.importance + employee.subordinates.stream()
                .mapToInt(this::getTotalImportance)
                .sum();
    }
}