题目描述:
给定一个保存员工信息的数据结构,它包含了员工唯一的id,重要度 和 直系下属的id。
比如,员工1是员工2的领导,员工2是员工3的领导。他们相应的重要度为15, 10, 5。那么员工1的数据结构是[1, 15, [2]],员工2的数据结构是[2, 10, [3]],员工3的数据结构是[3, 5, []]。注意虽然员工3也是员工1的一个下属,但是由于并不是直系下属,因此没有体现在员工1的数据结构中。
现在输入一个公司的所有员工信息,以及单个员工id,返回这个员工和他所有下属的重要度之和。
题目分析:
类似于图的深度遍历,每次从所要查询的员工id开始,求该员工的重要度和该员工所有下属的重要度之和。代码如下:
class Solution {
public:
int getImportance(vector<Employee*> employees, int id) {
Employee* e; //临时指向员工类的指针
for(int i=0;i<employees.size();i++){ //从员工表中进行遍历,查找要求的id
e=employees[i];
if(e->id==id){ //找到了目标员工的id
for(int j=0;j<e->subordinates.size();j++){ //遍历其所有的下属员工
e->importance+=getImportance(employees,e->subordinates[j]); //该员工的重要度加上所有下属员工的重要度即可
}
return e->importance; //求出结果之后,返回该员工的重要度
}
}
}
};