LC刷题专题：dfs、哈希表合集

自己刷题缺少分类思想，总是这里刷一道那里刷一道，以后建立几个专辑，然后自己新刷的同类型的题目都会即使更新上。

690. 员工的重要性

2024-10-03
题目描述：

我第一次写并没有考虑到dfs，首先使用一个哈希表构建 id 和 对应下属的映射关系。并且将此员工的中重要度加入到下述集合中，这样就不需要单独存储一个 Id 对应的重要度，直接根据map映射关系，根据id获取到对应下属映射，此后从下属映射的List集合的最后一个元素就是此员工对应的重要性。
上述操作完成后，联想到二叉树的层状遍历，直接构建了一个双端队列，队列里面存储员工id，之后循环队列即可完成操作。
详细代码如下：

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

class Solution {
    public int getImportance(List<Employee> employees, int id) {
        int ans = 0;
        HashMap<Integer, List<Integer>> map = new HashMap<>();
        for(Employee employ : employees){
            List<Integer> sub = employ.subordinates;
            if(sub == null){
                sub = new ArrayList<>();
            }
            sub.add(employ.importance);
            map.put(employ.id, sub);
        }
        Deque<Integer> dq = new ArrayDeque<>();
        dq.addLast(id);
        while(!dq.isEmpty()){
            int size = dq.size();
            while(size-- > 0){
                int idx = dq.pollFirst();
                List<Integer> temp = map.get(idx);
                ans += temp.get(temp.size() - 1);
                for(int i = 0; i < temp.size() - 1; i++){
                    dq.addLast(temp.get(i));
                }
            }
        }
        return ans;
    }
}

这种做法是使用迭代代替递归，参考二叉树的做法，上述也可以改写为递归版本的。这个版本参考了题解(灵茶山艾府)的一个版本，供参考：如果此前做过dfs以及二叉树，直接看下面代码非常易懂了。

class Solution {
    public int getImportance(List<Employee> employees, int id) {
        Map<Integer, Employee> employeeMap = new HashMap<>(employees.size());
        for (Employee e : employees) {
            employeeMap.put(e.id, e);
        }
        return dfs(employeeMap, id);
    }

    private int dfs(Map<Integer, Employee> employeeMap, int id) {
        Employee e = employeeMap.get(id);
        int res = e.importance;
        for (int subId : e.subordinates) {
            res += dfs(employeeMap, subId);
        }
        return res;
    }
}