2127.参加会议最多的员工数
题意:
解析:
建图, 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);
}
};