690. 员工的重要性

690. 员工的重要性

---

题目链接：690. 员工的重要性

代码如下：

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

class Solution 
{
public:
    int getImportance(vector<Employee*> employees, int id)
    {
        unordered_map<int, Employee*> um;
        for (auto& employe : employees)
        {
            um[employe->id] = employe;
        }

        function<int(int)> dfs = [&](int id)->int
            {
                int total = um[id]->importance;
                for (int subId : um[id]->subordinates)
                {
                    total += dfs(subId);
                }
                return total;
            };
        return dfs(id);
    }
};