690. Employee Importance

You have a data structure of employee information, including the employee’s unique ID, importance value, and direct subordinates’ IDs.

You are given an array of employees employees where:

• employees[i].id is the ID of the ith employee.
• employees[i].importance is the importance value of the ith employee.
• employees[i].subordinates is a list of the IDs of the direct subordinates of the ith employee.

Given an integer id that represents an employee’s ID, return the total importance value of this employee and all their direct and indirect subordinates.

/*
// Definition for Employee.
class Employee {
public:
    int id;
    int importance;
    vector<int> subordinates;
};
*/

class Solution {
    unordered_map<int, Employee*> employeeIdMap;
    int res = INT_MIN;
    int getTotalImportance(Employee* employee) {
        int importance = employee->importance;
        for(auto& sub : employee->subordinates) {
            importance += getTotalImportance(employeeIdMap[sub]);
        }
        return importance;
    }
public:
    int getImportance(vector<Employee*> employees, int id) {
        for(auto& e : employees)
            employeeIdMap[e->id] = e;
        return getTotalImportance(employeeIdMap[id]);
    }
};