LeetCode 5354. 通知所有员工所需的时间

5354. 通知所有员工所需的时间

公司里有 n 名员工，每个员工的 ID 都是独一无二的，编号从 0 到 n - 1。公司的总负责人通过 headID 进行标识。

在 manager 数组中，每个员工都有一个直属负责人，其中 manager[i] 是第 i 名员工的直属负责人。对于总负责人，manager[headID] = -1。题目保证从属关系可以用树结构显示。

公司总负责人想要向公司所有员工通告一条紧急消息。他将会首先通知他的直属下属们，然后由这些下属通知他们的下属，直到所有的员工都得知这条紧急消息。

第 i 名员工需要 informTime[i] 分钟来通知它的所有直属下属（也就是说在 informTime[i] 分钟后，他的所有直属下属都可以开始传播这一消息）。

返回通知所有员工这一紧急消息所需要的 分钟数 。

思路：忘了，比赛的时候写了出来。用dp自底向上

class Solution {
public:
    int numOfMinutes(int n, int headID, vector<int>& manager, vector<int>& informTime) {
        fill(dp, dp+n, 0);
        m.clear();
		for(int i=0;i<n;i++){
			if(manager[i]!=-1){
				m[manager[i]].push_back(i);
			}
		}
		return fun(n, headID, informTime);
    }
private:
	int dp[100005];
	unordered_map<int, vector<int> > m;
	int fun(int n, int nowID, vector<int>& informTime){
		if(informTime[nowID]==0){
			dp[nowID]=0;
			return dp[nowID];
		}
		
		vector<int> next_people = m[nowID];//nowID的下属
		
		int max_next_people_time=0;
		for(int i=0;i<next_people.size();i++){
			max_next_people_time = max(max_next_people_time, fun(n, next_people[i], informTime));
		} 
		dp[nowID] = max_next_people_time + informTime[nowID];
		return dp[nowID];
	}
};