员工的重要性

2024/8/26 力扣每日一题dfs

题目链接:https://leetcode.cn/problems/employee-importance/ (opens new window)

难度:中等

你有一个保存员工信息的数据结构,它包含了员工唯一的 id ,重要度和直系下属的 id 。

给定一个员工数组 employees,其中:

  • employees[i].id 是第 i 个员工的 ID。
  • employees[i].importance 是第 i 个员工的重要度。
  • employees[i].subordinates 是第 i 名员工的直接下属的 ID 列表。

给定一个整数 id 表示一个员工的 ID,返回这个员工和他所有下属的重要度的 总和。

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

示例 2:
输入:employees = [[1,2,[5]],[5,-3,[]]], id = 5
输出:-3
解释:员工 5 的重要度为 -3 并且没有直接下属。
因此,员工 5 的总重要度为 -3。

提示
1 <= employees.length <= 2000
1 <= employees[i].id <= 2000
所有的 employees[i].id 互不相同。
-100 <= employees[i].importance <= 100
一名员工最多有一名直接领导,并可能有多名下属。
employees[i].subordinates 中的 ID 都有效。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

这道中等题也比较简单,给的是一个对象数组,既然给了数组最好就不要想着去根据层级关系去建一颗树了,因为太浪费时间和空间了。

直接遍历一遍数组记录下对象id与对象的映射关系,然后根据目标对象的id,去计算他和他下属的重要度总和。

具体实现为深度优先搜索,当然也可以用队列来做广度优先,还是比较简单的,不像是中等题。

/**
 * Definition for Employee.
 * function Employee(id, importance, subordinates) {
 *     this.id = id;
 *     this.importance = importance;
 *     this.subordinates = subordinates;
 * }
 */

/**
 * @param {Employee[]} employees
 * @param {number} id
 * @return {number}
 */
var GetImportance = function(employees, id) {
  const ehash = new Map()
  employees.forEach(e => {
    ehash.set(e.id, e)
  })
  // 计算数组中所有员工的重要度(含下属)
  const getSubImp = (ids) => {
    let imp = 0
    for(let i = 0; i < ids.length; i++) {
      const emp = ehash.get(ids[i])
      imp += emp.importance + getSubImp(emp.subordinates)
    }
    return imp
  }

  return getSubImp([id])
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
Last Updated: 2024/9/4 15:42:44