Employee Importance Python Solution with Tests

lcpy gen -n 690   # by problem number
lcpy gen -s employee_importance   # by problem name

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

Input: employees = [[1,2,[5]],[5,-3,[]]], id = 5
Output: -3
Explanation: Employee 5 has an importance value of -3 and has no direct subordinates. Thus, the total importance value of employee 5 is -3.

class Employee:
    def __init__(self, id: int, importance: int, subordinates: list[int] | None = None) -> None:
        self.id = id
        self.importance = importance
        self.subordinates = subordinates if subordinates is not None else []


class Solution:
    # Time: O(n)
    # Space: O(n)
    def get_importance(self, employees: list[Employee], id: int) -> int:
        by_id = {employee.id: employee for employee in employees}
        total = 0
        stack = [id]
        while stack:
            employee = by_id[stack.pop()]
            total += employee.importance
            stack.extend(employee.subordinates)
        return total