跳转至

LeetCode: 690. 员工的重要性

1、题目描述

给定一个保存员工信息的数据结构,它包含了员工**唯一的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, 5, [2, 3]], [2, 3, []], [3, 3, []]], 1
输出: 11
解释:
员工1自身的重要度是5,他有两个直系下属2和3,而且2和3的重要度均为3。因此员工1的总重要度是 5 + 3 + 3 = 11。

注意:

  1. 一个员工最多有一个**直系**领导,但是可以有多个**直系**下属
  2. 员工数量不超过2000。

2、解题思路

​ 基本思路就是,建立一个哈希,或者说字典,保存id和里面内容的映射,不断的遍历属下,可以用递归,也可以使用list不断增加的特性,直到所有的都遍历完成

"""
# Employee info
class Employee:
    def __init__(self, id, importance, subordinates):
        # It's the unique id of each node.
        # unique id of this employee
        self.id = id
        # the importance value of this employee
        self.importance = importance
        # the id of direct subordinates
        self.subordinates = subordinates
"""
class Solution:
    em = {}
    def getImportance(self, employees, id):
        """
        :type employees: Employee
        :type id: int
        :rtype: int
        """


        for i in employees:
            self.em[i.id] = i
            # print(i)
        return self.getSum(self.em[id])

    def getSum(self,employee):
        total = employee.importance
        for i in employee.subordinates:
            total += self.getSum(self.em[i])
        return total
"""
# Employee info
class Employee:
    def __init__(self, id, importance, subordinates):
        # It's the unique id of each node.
        # unique id of this employee
        self.id = id
        # the importance value of this employee
        self.importance = importance
        # the id of direct subordinates
        self.subordinates = subordinates
"""
class Solution:

    def getImportance(self, employees, id):
        """
        :type employees: Employee
        :type id: int
        :rtype: int
        """

        em = {}
        for i in employees:
            em[i.id] = (i.importance, i.subordinates)

        sub = em[id][1]
        total = em[id][0];
        for i in sub:
            total += em[i][0]
            sub += em[i][1]


        return total