美文网首页
690. 员工的重要性

690. 员工的重要性

作者: 好吃红薯 | 来源:发表于2019-05-10 21:01 被阅读0次

现在输入一个公司的所有员工信息,以及单个员工id,返回这个员工和他所有下属的重要度之和。

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

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

"""
# Employee info
class Employee(object):
    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(object):
    
    def getImportance(self, employees, id):
        """
        :type employees: Employee
        :type id: int
        :rtype: int
        """
        self.res = 0
        hashmap = dict()
        for i in employees:
            hashmap[i.id] = [i.importance, i.subordinates]
            
        def dfs(subordinates):
            for j in subordinates:
                self.res += hashmap[j][0]
                dfs(hashmap[j][1])    
        dfs([id])        
        
        return self.res

相关文章

网友评论

      本文标题:690. 员工的重要性

      本文链接:https://www.haomeiwen.com/subject/rypzoqtx.html