【力扣每日一题 231101】LEETCODE 2127. 参加会议的最多员工数（基环内向树 拓扑排序）

引言

题意

$n$个员工围着圆桌坐，每个员工都有且仅有1个自己喜欢的员工，规定：只有和自己喜欢的员工挨着坐时，该员工才会出席会议，现在要最大化出席员工数，输出最多员工数。

基环树

所谓基环树，其实不是树，就像鲸鱼并不是鱼、河马不是马、冰糖葫芦不是葫芦一样。之所以叫树，是因为它只比树多一条边，即$n$个结点、$n$条边的图称做基环树，不难推出它一定含有环。
直观地说：根据与环相连的链，如果区分有向边，且都是向外延伸的，即为基环外向树，反之为基环内向树。更严格的说，如果每个结点的出度都为1，成为基环内向树，就像本道例题一样。

例题

【LEETCODE 2127】

算法描述

本题的关键在于找到最优策略，针对环的大小，尝试之后可以发现分为两种情况：
① 若环的大小大于等于3，那么要取这个环的话，不能再取其他结点；
② 若环的大小等于2（即两个人互相喜欢…），那么这是个百搭的方案，有与其中任何一个结点相连的链都可以加进来（但是一个结点只能算一条链，即与之相连的最长链），并且如果有其他的大小为2的环也可以进行组合，都相加统计进答案里。

有了上述分析，图论化的解题描述如下：
方案一：找出最长的环，答案为最长的环的大小；
方案二：找出所有大小为2的环，针对所有这些环的结点，找出以这些结点为终点的最长链，把最长链累加即为答案。
最后，把两个方案的答案取MAX即为最终答案。

具体的代码操作：
方案一：根据边的指向进行遍历，如果遇到指向的点和之前已经遍历的点重复，即找到了环，算出环的长度即可，最后取最大值找出最长的环的大小。
方案二：算出以每个结点为终点的最长的链的大小，用拓扑排序去求，只需要略微修改拓扑排序的代码即可，更新最长的链的大小。然后遍历找出在 大小为2的环 中的结点，将前面算得的最长的链的大小求和。
最后取最大值。

代码实现

class Solution {
public:
    int maximumInvitations(vector<int>& favorite) {
        return max(get_max_cycle(favorite), get_topo(favorite));
    }

    int get_max_cycle(vector<int>& v) {
        int ret = -1;
        int n = v.size();
        vector<bool> vis(n);
        for (int i = 0; i < n; i++) {
            if (vis[i]) {
                continue;
            }
            int j = i;
            vector<int> cycle;
            while (!vis[j]) {
                cycle.push_back(j);
                vis[j] = 1;
                j = v[j];
            }
            int tmp_n = cycle.size();
            for (auto k : cycle) {
                if (k == j) {
                    break;
                }
                --tmp_n;
            }
            ret = max(ret, tmp_n);
        }
        return ret;
    }

    int get_topo(vector<int>& a) {
        int ret = 0;
        int n = a.size();
        vector<int> du(n);
        vector<int> chain(n);
        queue<int> q;
        for (auto i : a) {
            du[i]++;
        }
        for (int i = 0; i < n; i++) {
            chain[i] = 1;
            if (du[i] == 0) {
                q.push(i);
            }
        }
        while (!q.empty()) {
            int fro = q.front();
            q.pop();
            chain[a[fro]] = chain[fro] + 1;
            du[a[fro]]--;
            if (du[a[fro]] == 0) {
                q.push(a[fro]);
            }
        }
        for (int i = 0; i < n; i++) {
            if (a[a[i]] == i) {
                ret += chain[i];
            }
        }
        return ret;
    }
};

关键词

力扣 每日一题 参加会议的最多员工数 2127 基环内向树 拓扑排序