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.
Example 1:
Input: employees = [[1,5,[2,3]],[2,3,[]],[3,3,[]]], id = 1
Output: 11
Explanation: Employee 1 has an importance value of 5 and has two direct subordinates: employee 2 and employee 3.
They both have an importance value of 3.
Thus, the total importance value of employee 1 is 5 + 3 + 3 = 11.
Example 2:
Input: employees = [[1,2,[5]],[5,-3,[]]], id = 5
Output: -3
Explanation: Employee 5 has an importance value of -3 and has no direct subordinates.
Thus, the total importance value of employee 5 is -3.
Constraints:
1 <= employees.length <= 2000
1 <= employees[i].id <= 2000
All employees[i].id are unique.
-100 <= employees[i].importance <= 100
One employee has at most one direct leader and may have several subordinates.
The IDs in employees[i].subordinates are valid IDs.
Solutions
Solution 1: Hash Table + DFS
We use a hash table $d$ to store all employee information, where the key is the employee's ID, and the value is the employee object. Then we start a depth-first search from the given employee ID. Each time we traverse to an employee, we add the employee's importance to the answer, and recursively traverse all the subordinates of the employee, adding the importance of the subordinates to the answer as well.
The time complexity is $O(n)$, and the space complexity is $O(n)$. Where $n$ is the number of employees.
/*// Definition for Employee.class Employee { public int id; public int importance; public List<Integer> subordinates;};*/classSolution{privatefinalMap<Integer,Employee>d=newHashMap<>();publicintgetImportance(List<Employee>employees,intid){for(vare:employees){d.put(e.id,e);}returndfs(id);}privateintdfs(inti){Employeee=d.get(i);ints=e.importance;for(intj:e.subordinates){s+=dfs(j);}returns;}}
/*// Definition for Employee.class Employee {public: int id; int importance; vector<int> subordinates;};*/classSolution{public:intgetImportance(vector<Employee*>employees,intid){unordered_map<int,Employee*>d;for(auto&e:employees){d[e->id]=e;}function<int(int)>dfs=[&](inti)->int{ints=d[i]->importance;for(intj:d[i]->subordinates){s+=dfs(j);}returns;};returndfs(id);}};
1 2 3 4 5 6 7 8 9101112131415161718192021222324
/** * Definition for Employee. * type Employee struct { * Id int * Importance int * Subordinates []int * } */funcgetImportance(employees[]*Employee,idint)int{d:=map[int]*Employee{}for_,e:=rangeemployees{d[e.Id]=e}vardfsfunc(int)intdfs=func(iint)int{s:=d[i].Importancefor_,j:=ranged[i].Subordinates{s+=dfs(j)}returns}returndfs(id)}