本文共 1226 字，大约阅读时间需要 4 分钟。

为了解决这个问题，我们需要找到某个员工及其所有直系下属的重要度之和。我们可以将员工信息看作是一个树形结构，其中每个员工是一个结点，直系下属是子节点。我们可以使用广度优先搜索（BFS）来高效地遍历整个子树，计算所有重要度之和。

方法思路

解决代码

import sysfrom collections import dequedef get_importance(employees, target_id):    # 创建邻接表    table = {}    for employee in employees:        employee_id = employee[0]        importance = employee[1]        subordinates = employee[2]        table[employee_id] = (importance, subordinates)        # 初始化队列并设置结果    queue = deque([target_id])    total_importance = 0        while queue:        current_id = queue.popleft()        imp, subs = table[current_id]        total_importance += imp        # 将下属加入队列        queue.extend(subs)        return total_importanceif __name__ == "__main__":    # 示例输入    input_employees = [        [1, 5, [2, 3]],        [2, 3, []],        [3, 3, []]    ]    target_id = 1    result = get_importance(input_employees, target_id)    print(result)

代码解释

这种方法确保了我们能够高效地遍历整个子树，计算所有重要度之和。BFS的时间复杂度为O(N)，其中N是员工数量，适用于处理较大的数据量。

转载地址：http://jscr.baihongyu.com/