题目
给定一个保存员工信息的数据结构,它包含了员工唯一的id,重要度 和 直系下属的id。
比如,员工1是员工2的领导,员工2是员工3的领导。他们相应的重要度为15, 10, 5。那么员工1的数据结构是[1, 15, [2]],员工2的数据结构是[2, 10, [3]],员工3的数据结构是[3, 5, []]。注意虽然员工3也是员工1的一个下属,但是由于并不是直系下属,因此没有体现在员工1的数据结构中。
现在输入一个公司的所有员工信息,以及单个员工id,返回这个员工和他所有下属的重要度之和。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/employee-importance
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
1.先将所给id入队列,在队列不为空的前提下,在list中找到员工id与所给id相同的员工记为老员工 ,将其重要性记录进行后续累加。
2.找到老员工的下属员工,将下属员工入队列
3.通过while循环,依次更新每次id的值,再将下属员工的重要性累加起来并最终返回结果。
代码
package homeworks;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
public class List1 {
class Employee {
public int id;
public int importance;
public List<Integer> subordinates;
}
//员工的重要性
class Solution {
public int getImportance(List<Employee> employees, int id) {
//1.当员工id与输入id相同时,将此员工的重要性入队列,再出队列,记录其重要性
//2.再将当前员工的下属员工的重要性依次入队列,再出队列,将重要性累加求和
Queue<Integer> queue = new LinkedList<>();
if (employees == null){
return 0;
}
int result = 0;
queue.offer(id);
while (!queue.isEmpty()){
int temp = queue.poll();//temp里面存的就是当前id值
//遍历员工list找到某个员工的id与当前temp值相同即为所求员工
//内部循环是将下属入队列出队列操作
//外部循环是累加重要性
for (Employee e:employees){
if (e.id == temp ) {
//说明所求员工就是他,将他的重要性相加起来
result += e.importance;
//再依次将他的下属员工入队列,再出队列,累加重要性
for (Integer i : e.subordinates) {
if (i == null) {
//代表没有下属了,循环也就结束了
return result;
}
queue.offer(i);
}
}
}
}
return result;
}
}
}