题目地址:
https://leetcode.com/problems/maximum-employees-to-be-invited-to-a-meeting/description/
有 n n n个人编号 0 ∼ n − 1 0\sim n-1 0∼n−1,给定一个长 n n n数组 a a a, a [ i ] a[i] a[i]表示 i i i号人喜欢的人的编号(他不能喜欢自己)。有一个圆桌,想安排若干人上桌,每个人必须要和自己喜欢的人坐一起。问最多能有多少人上桌。
以每个人为点建图,当且仅当 a a a喜欢 b b b的时候 a a a和 b b b之间有一条边。我们首先考虑这个图是个什么样的结构。首先任意选择一个点出发,它有且仅有一条出边,所以顺着这条边向下走,一定能走到一个环上。当然特殊情况是这个点本身就在环上。我们定义一个环上的点的“粉丝团“这样的概念。找一个入度为 0 0 0的点 u u u,那么它一定能沿着边走到环上的某个点 v v v(假设 v v v是第一个走到的环上的点),那么从 u u u走到 v v v这条链上的除了点 v v v之外的所有点,就称为点 v v v的“粉丝团”。容易知道对于任意一个环上的点,它可能有粉丝团,也可能没有,即使有,也可能有多个。这个图事实上可以分解为若干个不相交的子图,使得每个子图要么是环,要么是环加上环上每个点的所有粉丝团。
现在我们考虑如何上桌。对于长度大于 2 2 2的环,他们只能单独上桌,因为他们之中只要有一个人上桌,那每个人的左右两边的人都确定了。对于长度等于 2 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 ) O(n) O(n)。