博客
关于我
力扣 690. 员工的重要性
阅读量:353 次
发布时间:2019-03-04

本文共 2175 字,大约阅读时间需要 7 分钟。

题目描述

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

提示:

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

来源:力扣(LeetCode)

链接:
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


BFS

其实这道题的数据结构可以理解为树形结构:

  • 员工是树的结点
  • 员工的重要度是结点的值
  • 直系领导是父结点,直系下属是子结点

于是问题就转化为求以某员工为根结点形成的子树的所有结点值之和,可以使用 BFSDFS 两种思路求解。

另外需要对输入数据进行预处理,用字典以邻接表的形式存储树形结构,这样可以快速查询某员工信息。

下面是 BFS 求解过程:

"""# 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:        # 邻接表        table = {   }        for e in employees:            table[e.id] = (e.importance, e.subordinates)        # BFS        from collections import deque        dq = deque([id])        res = 0        while dq:            imp, sub = table[dq.popleft()]            res += imp            dq.extend(sub)        return res

运行结果:

执行结果:通过

执行用时:164 ms, 在所有 Python3 提交中击败了69.60% 的用户
内存消耗:16.2 MB, 在所有 Python3 提交中击败了8.60% 的用户


DFS

下面是 DFS 求解过程:

"""# 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:        # 邻接表        table = {   }        for e in employees:            table[e.id] = (e.importance, e.subordinates)        # DFS        def dfs(root):            res = 0            imp, sub = table[root]            res += imp            for s in sub:                res += dfs(s)            return res        return dfs(id)

运行结果:

执行结果:通过

执行用时:156 ms, 在所有 Python3 提交中击败了91.76% 的用户
内存消耗:16.2 MB, 在所有 Python3 提交中击败了10.99% 的用户


你可能感兴趣的文章
mysql 如何给SQL添加索引
查看>>
mysql 字段区分大小写
查看>>
mysql 字段合并问题(group_concat)
查看>>
mysql 字段类型类型
查看>>
MySQL 字符串截取函数,字段截取,字符串截取
查看>>
MySQL 存储引擎
查看>>
mysql 存储过程 注入_mysql 视图 事务 存储过程 SQL注入
查看>>
MySQL 存储过程参数:in、out、inout
查看>>
mysql 存储过程每隔一段时间执行一次
查看>>
mysql 存在update不存在insert
查看>>
Mysql 学习总结(86)—— Mysql 的 JSON 数据类型正确使用姿势
查看>>
Mysql 学习总结(87)—— Mysql 执行计划(Explain)再总结
查看>>
Mysql 学习总结(88)—— Mysql 官方为什么不推荐用雪花 id 和 uuid 做 MySQL 主键
查看>>
Mysql 学习总结(89)—— Mysql 库表容量统计
查看>>
mysql 实现主从复制/主从同步
查看>>
mysql 审核_审核MySQL数据库上的登录
查看>>
mysql 导入 sql 文件时 ERROR 1046 (3D000) no database selected 错误的解决
查看>>
mysql 导入导出大文件
查看>>
MySQL 导出数据
查看>>
mysql 将null转代为0
查看>>