树
深度优先搜索
广度优先搜索
数组
哈希表
题目描述
给定一个保存员工信息的数据结构,它包含了员工 唯一的 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 。
解法
方法一
Python3 Java JavaScript
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 """
# 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 :
m = { emp . id : emp for emp in employees }
def dfs ( id : int ) -> int :
emp = m [ id ]
s = emp . importance
for sub in emp . subordinates :
s += dfs ( sub )
return s
return dfs ( 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 /*
// Definition for Employee.
class Employee {
public int id;
public int importance;
public List<Integer> subordinates;
};
*/
class Solution {
private final Map < Integer , Employee > map = new HashMap <> ();
public int getImportance ( List < Employee > employees , int id ) {
for ( Employee employee : employees ) {
map . put ( employee . id , employee );
}
return dfs ( id );
}
private int dfs ( int id ) {
Employee employee = map . get ( id );
int sum = employee . importance ;
for ( Integer subordinate : employee . subordinates ) {
sum += dfs ( subordinate );
}
return sum ;
}
}
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 /**
* 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 map = new Map ();
for ( const employee of employees ) {
map . set ( employee . id , employee );
}
const dfs = id => {
const employee = map . get ( id );
let sum = employee . importance ;
for ( const subId of employee . subordinates ) {
sum += dfs ( subId );
}
return sum ;
};
return dfs ( id );
};
GitHub