dfs 力扣630 比较绕的一道题

最新推荐文章于 2024-05-09 22:59:19 发布

lucky tiger

最新推荐文章于 2024-05-09 22:59:19 发布

阅读量111

点赞数

分类专栏： leetcode力扣 # DFS

本文链接：https://blog.csdn.net/weixin_42269817/article/details/105908313

版权

leetcode力扣同时被 2 个专栏收录

67 篇文章 0 订阅

订阅专栏

DFS

11 篇文章 0 订阅

订阅专栏

1、

/*
// Definition for Employee.
class Employee {
public:
    int id;
    int importance;
    vector<int> subordinates;
};
*/

class Solution {
public:
    int getImportance(vector<Employee*> employees, int id) {
        unordered_map<int,Employee*> mp;
        for(auto employee:employees)
        {
            mp[employee->id]=employee;
        }
        return dfs(mp,id);
    }
    int dfs(unordered_map<int, Employee*> mp, int id)
    {
        if(mp.find(id)==mp.end())
        {
            return 0;
        }
        queue<Employee*> que;
        que.push(mp[id]);
        int ans=0;
        while(!que.empty())
        {
            Employee* curr=que.front();
            que.pop();
            ans+=curr->importance;

            for(auto sub:curr->subordinates)
            {
                que.push(mp[sub]);
            }
        }
        return ans;
    }
};