1376. 通知所有员工所需的时间
公司里有 n 名员工,每个员工的 ID 都是独一无二的,编号从 0 到 n - 1。公司的总负责人通过 headID 进行标识。
在 manager 数组中,每个员工都有一个直属负责人,其中 manager[i] 是第 i 名员工的直属负责人。对于总负责人,manager[headID] = -1。题目保证从属关系可以用树结构显示。
公司总负责人想要向公司所有员工通告一条紧急消息。他将会首先通知他的直属下属们,然后由这些下属通知他们的下属,直到所有的员工都得知这条紧急消息。
第 i 名员工需要 informTime[i] 分钟来通知它的所有直属下属(也就是说在 informTime[i] 分钟后,他的所有直属下属都可以开始传播这一消息)。
返回通知所有员工这一紧急消息所需要的 分钟数 。
const int MAXN = 1e5+10;
int dict[MAXN];
vector<int> edge[MAXN];
class Solution {
public:
int ans = 0;
int numOfMinutes(int n, int headID, vector<int>& manager, vector<int>& informTime) {
ans = 0;
for (int i = 0;i<n;i++){
dict[i]=0;
edge[i].clear();
}
for (int i =0;i<n;i++){
if(manager[i]!=-1){
edge[manager[i]].push_back(i);
}
}
queue<int> q;
q.push(headID);
dict[headID]=0;
while(!q.empty()){
int x = q.front();
q.pop();
ans = max(ans,dict[x]);
for (auto v:edge[x]){
dict[v] = dict[x]+informTime[x];
q.push(v);
}
}
return ans;
}
};