【LeetCode】第690题——员工的重要性（难度：简单）

题目描述

给定一个保存员工信息的数据结构，它包含了员工唯一的id，重要度 和 直系下属的id。

比如，员工1是员工2的领导，员工2是员工3的领导。他们相应的重要度为15, 10, 5。那么员工1的数据结构是[1, 15, [2]]，员工2的数据结构是[2, 10, [3]]，员工3的数据结构是[3, 5, []]。注意虽然员工3也是员工1的一个下属，但是由于并不是直系下属，因此没有体现在员工1的数据结构中。

现在输入一个公司的所有员工信息，以及单个员工id，返回这个员工和他所有下属 （不只是他的直系下属） 的重要度之和。

1. 示例 1:
   输入: [[1, 5, [2, 3]], [2, 3, []], [3, 3, []]], 1
   输出: 11
   解释:员工1自身的重要度是5，他有两个直系下属2和3，而且2和3的重要度均为3。因此员工1的总重要度是 5 + 3 + 3 = 11。

注意:
一个员工最多有一个直系领导，但是可以有多个直系下属
员工数量不超过2000。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/employee-importance
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

因为List< Employee >难以按id进行快速索引，因此使用HashMap<Integer, Employee>键值对来进行索引。

思路一：递归。对id号员工的下属进行递归，每次递归均把重要性累加起来。即下属套下属，然后求重要性的累加和。

思路二：栈。先把id号员工压入栈，然后弹出栈顶并记录，再把弹出元素对应的下属依次压入栈，每次弹栈均要把重要性先进行累加。

代码详解

思路一：递归

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

class Solution {

    int ans = 0;	// 结果
    HashMap<Integer, Employee> hm;	// HashMap
    
    public int getImportance(List<Employee> employees, int id) {
        hm = new HashMap<>();
        for(Employee emp : employees) {
            hm.put(emp.id, emp);
        }
        // 上面是构建键值对
        // 下面是递归本体，需要id号和Employee作为参数
        func(id, hm.get(id));
        return ans;
    }

    public void func(int id, Employee emp) {
        ans += hm.get(id).importance;	// 先对重要性进行累加
        for(int num : emp.subordinates) {
            func(num, hm.get(num));		// 然后把他的下属也进行递归，如此一来，下属的下属也可以进入递归中
        }
    }
}

思路二：栈

class Solution {
    public int getImportance(List<Employee> employees, int id) {
        int ans = 0;	// 结果
        // 构建键值对
        HashMap<Integer, Employee> hm = new HashMap<>();
        for(Employee emp : employees) {
            hm.put(emp.id, emp);
        }
        Stack<Integer> stack = new Stack<>();
        stack.push(id);	// 栈直接压入id即可，不用压入Employee
        while(!stack.isEmpty()) {	// 循环终止条件是栈空
            int pop = stack.pop();	// 获取栈顶元素并弹出
            ans += hm.get(pop).importance;	// 对重要性进行累加
            for(int num : hm.get(pop).subordinates) {
                stack.push(num);	// 把弹出id号的下属依次压入栈，如此一来，下属的下属也可以压入栈
            }
        }
        return ans;
    }
}

注意点

1. 本题可以说是树的变体，不要被唬住，实际思路与代码都很简单。
2. 要学会建立索引来降低时间复杂度。