Saved from https://leetcode.com/problems/employee-importance/

lc690

"""
# 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:
        id_map = { e.id: e for e in employees }
        def aggregate(id: int) -> int:
            e = id_map[id]
            return e.importance + sum(aggregate(sub_id) for sub_id in e.subordinates)
        return aggregate(id)