leetcode 2127.参加会议最多的员工数

2127.参加会议最多的员工数

题意:

leetcode 2127.参加会议最多的员工数_第1张图片

解析:

建图, i i i f a v o r i t e i favorite_i favoritei 连边,图是内向基环森林。

  • 环长度为 2 的基环树可以放置多棵,节点数为每棵树节点数的和
  • 环长度 > 2 的基环树只能放置一棵,节点数为所有树的环长度的最大值

所以可以通过拓扑排序使基环树只剩下环,然后就可以求出每个环的长度,因此情况2可以实现;拓扑排序的时候顺便建立反图,然后就可以求出最大深度,情况1可以实现

代码:

class Solution {
public:
    int maximumInvitations(vector<int>& favorite) {
        int n = favorite.size();
        vector<int> degin(n, 0);
        vector<vector<int> >g(n); 
        for(int i = 0; i < n; i++)
            degin[favorite[i]]++;
        queue<int> q;
        for(int i = 0; i < n; i++){
            if(degin[i] == 0)
                q.push(i);
        }
        while(q.size()){
            int u = q.front();
            q.pop();
            int v = favorite[u];
            g[v].push_back(u);
            if(--degin[v] == 0)
                q.push(v);
        }
        // 返回最大深度
        function<int(int)> dfs1 = [&](int u) -> int{
            int dep = 1;
            for(auto v : g[u])
                dep = max(dep, dfs1(v)+1);
            return dep;
        };
        // 返回环长度
        function<int(int)> circle = [&](int u) -> int{
            int len = 1;
            degin[u] = 0;
            int t = favorite[u];
            while(t != u){
                len++;
                degin[t] = 0;
                t = favorite[t];
            }
            return len;
        };
        int sum2 = 0, maxx = 0;
        for(int i = 0; i < n; i++){
            if(degin[i] == 0)
                continue;
            int len = circle(i);
            if(len == 2)
                sum2 += dfs1(favorite[i]) + dfs1(i);
            else
                maxx = max(maxx, circle(i));
        }
        return max(sum2, maxx);
    }
};

你可能感兴趣的:(leetcode每日一题,leetcode,算法,图论,数据结构)