LeetCode 690 员工的重要性[DFS Map] HERODING的LeetCode之路

Posted 2021-05-15 HERODING23

解题思路：
典型的DFS题目，核心思想是不断深度搜索遍历单个节点，一直到没有下属为止，注意为了方便操作，用Map保存 id 与 Employee的映射，代码如下：

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

class Solution {
private:
    unordered_map<int, Employee*> mp;
public:
    int dfs(int id) {
        Employee* emp = mp[id];
        int total = emp -> importance;
        for(int i = 0; i < emp->subordinates.size(); i ++) {
            total += dfs(emp->subordinates[i]);
        }
        return total;
    }

    int getImportance(vector<Employee*> employees, int id) {
        for(auto employee : employees) {
            mp[employee -> id] = employee;
        }
        return dfs(id);
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/employee-importance/solution/cmapdfs-by-heroding-9hy8/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/