【Leetcode】2127. Maximum Employees to Be Invited to a Meeting

题目地址：

https://leetcode.com/problems/maximum-employees-to-be-invited-to-a-meeting/description/

有$n$个人编号$0\sim n-1$，给定一个长$n$数组$a$，$a[i]$表示$i$号人喜欢的人的编号（他不能喜欢自己）。有一个圆桌，想安排若干人上桌，每个人必须要和自己喜欢的人坐一起。问最多能有多少人上桌。

以每个人为点建图，当且仅当$a$喜欢$b$的时候$a$和$b$之间有一条边。我们首先考虑这个图是个什么样的结构。首先任意选择一个点出发，它有且仅有一条出边，所以顺着这条边向下走，一定能走到一个环上。当然特殊情况是这个点本身就在环上。我们定义一个环上的点的“粉丝团“这样的概念。找一个入度为$0$的点$u$，那么它一定能沿着边走到环上的某个点$v$（假设$v$是第一个走到的环上的点），那么从$u$走到$v$这条链上的除了点$v$之外的所有点，就称为点$v$的“粉丝团”。容易知道对于任意一个环上的点，它可能有粉丝团，也可能没有，即使有，也可能有多个。这个图事实上可以分解为若干个不相交的子图，使得每个子图要么是环，要么是环加上环上每个点的所有粉丝团。

现在我们考虑如何上桌。对于长度大于$2$的环，他们只能单独上桌，因为他们之中只要有一个人上桌，那每个人的左右两边的人都确定了。对于长度等于$2$的环，这些环可以都上桌，并且带上两个人各自的人数最多的粉丝团上桌。

求每个环上的点的粉丝团的过程可以用拓扑排序来做。另外可以开一个数组存一下每个环都有哪些人。代码如下：

class Solution {
 public:
  int maximumInvitations(vector<int>& fs) {
    int n = fs.size();
    int ind[n];
    memset(ind, 0, sizeof ind);
    for (int i = 0; i < n; i++) ind[fs[i]]++;
    queue<int> q;
    for (int i = 0; i < n; i++) if (!ind[i]) q.push(i);
    // 对环上的点，存每个点的粉丝团的人数
    int len[n];
    memset(len, 0, sizeof len);
    bool vis[n];
    memset(vis, 0, sizeof vis);
    // 拓扑排序求环上的点的最大粉丝团人数，并筛选出环上的点是哪些
    while (q.size()) {
      int t = q.front(); q.pop();
      vis[t] = true;
      int ne = fs[t];
      len[ne] = max(len[ne], len[t] + 1);
      if (!--ind[ne]) q.push(ne);
    }

	int res = 0;
	// c2存长度为2的环有哪些
    vector<vector<int>> c2;
    for (int i = 0; i < n; i++)
      // 遍历环上的点
      if (!vis[i]) {
        // 求一下环上有几个点
        int cnt = 1;
        vector<int> v;
        vis[i] = true;
        v.push_back(i);
        for (int x = i; fs[x] != i; x = fs[x]) {
          v.push_back(fs[x]);
          vis[fs[x]] = true;
          cnt++;
        }
        // 如果环上只有2个点，那么把这个环存下来；否则的话这个环只能单独上桌，更新答案
        if (cnt == 2) c2.push_back(v);
        else res = max(res, cnt);
      }

	// 将所有2个点的环上桌，并且把每个人的最大的粉丝团也上桌
    int cnt = 2 * c2.size();
    for (auto& v : c2) cnt += len[v[0]] + len[v[1]];
    
    res = max(res, cnt);
    return res;
  }
};

时空复杂度$O(n)$。