LeetCode 690. Employee Importance

原题链接在这里：https://leetcode.com/problems/employee-importance/description/

题目：

You are given a data structure of employee information, which includes the employee's unique id, his importance value and his directsubordinates' id.

For example, employee 1 is the leader of employee 2, and employee 2 is the leader of employee 3. They have importance value 15, 10 and 5, respectively. Then employee 1 has a data structure like [1, 15, [2]], and employee 2 has [2, 10, [3]], and employee 3 has [3, 5, []]. Note that although employee 3 is also a subordinate of employee 1, the relationship is not direct.

Now given the employee information of a company, and an employee id, you need to return the total importance value of this employee and all his subordinates.

Example 1:

Input: [[1, 5, [2, 3]], [2, 3, []], [3, 3, []]], 1
Output: 11
Explanation:
Employee 1 has importance value 5, and he has two direct subordinates: employee 2 and employee 3. They both have importance value 3. So the total importance value of employee 1 is 5 + 3 + 3 = 11.

Note:

1. One employee has at most one direct leader and may have several subordinates.
2. The maximum number of employees won't exceed 2000.

题解：

类似Clone Graph, Nested List Weight Sum.

可以采用BFS. 每层的Employee挨个加进去.

Time Complexity: O(n). n是root id的所有下属个数, 包括直系下属和非直系下属.

Space: O(employees.size()). 全部员工生成的map.

AC Java:

/*
// Employee info
class Employee {
    // It's the unique id of each node;
    // unique id of this employee
    public int id;
    // the importance value of this employee
    public int importance;
    // the id of direct subordinates
    public List<Integer> subordinates;
};
*/
class Solution {
    public int getImportance(List<Employee> employees, int id) {
        int res = 0;
        
        HashMap<Integer, Employee> hm = new HashMap<Integer, Employee>();
        for(Employee employee : employees){
            hm.put(employee.id, employee);
        }
        
        LinkedList<Employee> que = new LinkedList<Employee>();
        que.add(hm.get(id));
        while(!que.isEmpty()){
            Employee cur = que.poll();
            res += cur.importance;
            for(int subId : cur.subordinates){
                que.add(hm.get(subId));
            }
        }
        
        return res;
    }
}

AC Python:

"""
# Definition for Employee.
class Employee:
    def __init__(self, id: int, importance: int, subordinates: List[int]):
        self.id = id
        self.importance = importance
        self.subordinates = subordinates
"""

class Solution:
    def getImportance(self, employees: List['Employee'], id: int) -> int:
        hm = {e.id : e for e in employees}
        que = deque([id])
        res = 0
        while que:
            cur = hm[que.popleft()]
            res += cur.importance
            que.extend(cur.subordinates)
        return res
        

DFS.逐层往深dfs.

Time Complexity: O(n). n是root id的所有下属个数, 包括直系下属和非直系下属.

Space: O(employees.size()). 全部员工生成的map.

AC Java:

/*
// Employee info
class Employee {
    // It's the unique id of each node;
    // unique id of this employee
    public int id;
    // the importance value of this employee
    public int importance;
    // the id of direct subordinates
    public List<Integer> subordinates;
};
*/
class Solution {
    public int getImportance(List<Employee> employees, int id) {
        HashMap<Integer, Employee> hm = new HashMap<Integer, Employee>();
        for(Employee employee : employees){
            hm.put(employee.id, employee);
        }
        
        return dfs(hm, id);
    }
    
    private int dfs(HashMap<Integer, Employee> hm, int id){
        int res = 0;
        Employee cur = hm.get(id);
        
        res += cur.importance;
        for(int subId : cur.subordinates){
            res += dfs(hm, subId);
        }
        return res;
    }
}

AC Python:

"""
# Definition for Employee.
class Employee:
    def __init__(self, id: int, importance: int, subordinates: List[int]):
        self.id = id
        self.importance = importance
        self.subordinates = subordinates
"""

class Solution:
    def getImportance(self, employees: List['Employee'], id: int) -> int:
        hm = {e.id : e for e in employees}

        def dfs(hm, id):
            cur = hm[id]
            res = cur.importance
            for next in cur.subordinates:
                res += dfs(hm, next)
            return res
        
        return dfs(hm, id)