博客
关于我
力扣 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% 的用户


你可能感兴趣的文章
MSSQL日期格式转换函数(使用CONVERT)
查看>>
MSTP多生成树协议(第二课)
查看>>
MSTP是什么?有哪些专有名词?
查看>>
Mstsc 远程桌面链接 And 网络映射
查看>>
Myeclipse常用快捷键
查看>>
MyEclipse更改项目名web发布名字不改问题
查看>>
MyEclipse用(JDBC)连接SQL出现的问题~
查看>>
mt-datetime-picker type="date" 时间格式 bug
查看>>
myeclipse的新建severlet不见解决方法
查看>>
MyEclipse设置当前行背景颜色、选中单词前景色、背景色
查看>>
Mtab书签导航程序 LinkStore/getIcon SQL注入漏洞复现
查看>>
myeclipse配置springmvc教程
查看>>
MyEclipse配置SVN
查看>>
MTCNN 人脸检测
查看>>
MyEcplise中SpringBoot怎样定制启动banner?
查看>>
MyPython
查看>>
MTD技术介绍
查看>>
MySQL
查看>>
MySQL
查看>>
mysql
查看>>