630员工的重要性
劳动节搞这种题一开就是故意的吧
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/employee-importance
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目
给定一个保存员工信息的数据结构,它包含了员工 唯一的 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 。
思路
-
一开始想到了匹配如果id和员工表中的id一致,就去遍历他的下属表,但是发现下属的下属也是它的下属,导致整道题解法出错了
-
- 然后就想到用队列,先把当前员工的id为传入的id的元素存进队列里
- 当队列不为空的时候,依次把队头元素取出来
- sum的值=原值+队头元素的importance的值,然后把队头pop出来
- 接着遍历队头的下属数组,依次把对应的下属的元素存进去
- 当队列为空的时候,遍历完了所有下属的importance值
代码
/*
// Definition for Employee.
class Employee {
public:
int id;
int importance;
vector<int> subordinates;
};
*/
class Solution {
public:
int getImportance(vector<Employee*> employees, int id) {
int sum=0;
queue<Employee*> que;
for(int i=0;i<employees.size();i++)
{
if(employees[i]->id==id)
{
que.push(employees[i]);
}
}
while(que.size()!=0)
{
Employee* temp=que.front();
que.pop();
sum=sum+temp->importance;
for(int i=0;i<temp->subordinates.size();i++)
{
for(int j=0;j<employees.size();j++)
{
if(employees[j]->id==temp->subordinates[i])
{
que.push(employees[j]);
}
}
}
}
return sum;
}
};