LeetCode 5354. 通知所有员工所需的时间(迭代)

 

题目链接：https://leetcode-cn.com/problems/time-needed-to-inform-all-employees/

       解法不一，可以从上向下的遍历，也可以从下往上遍历，可以深搜，也可以迭代，比赛的时候去实现的从下网上去查找，但是写复杂了，写了很多不必要的东西，其实这道题很简单。我们找到每一个叶子节点，然后从下网上搜到根节点就好了，代码也很好理解，直接看代码吧。

---

AC代码：

class Solution {
public:
    int numOfMinutes(int n, int root, vector<int>& m, vector<int>& info) {
        int ans = 0;
        for(int i=0;i<n;i++){
            if(info[i] == 0){      // 判断是否是叶子节点
                int xx = i;
                int sum = 0;
                while(xx != -1){
                    sum += info[xx];
                    xx = m[xx];
                }
                ans = max(ans, sum);
            }
        }
        return ans;
    }
};