题目描述
班里N个小朋友,每个人都有自己最崇拜的一个小朋友(也可以是自己)。
在一个游戏中,需要小朋友坐一个圈,
每个小朋友都有自己最崇拜的小朋友在他的右手边。
求满足条件的圈最大多少人?
小朋友编号为1,2,3,…N
输入第一行,一个整数N(3<N<100000)
接下来一行N个整数,由空格分开。
要求输出一个整数,表示满足条件的最大圈的人数。
测试样例
-
输入:
9
3 4 2 5 3 8 4 6 9 -
则程序应该输出:
4 -
输入:
30
22 28 16 6 27 21 30 1 29 10 9 14 24 11 7 2 8 5 26 4 12 3 25 18 20 19 23 17 13 15 -
程序应该输出:
16
解题思路
每个小朋友都有一个最崇拜的人,然后求这个多个崇拜圈的最大人数。有向图找最大环,用拓扑排序预处理一下把入度为0的人check掉,这些都是不能在圈中的,剩下的必然是存在于某个圈中。dfs求每个圈的人数更新最大值。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <queue>
using namespace std;
const int maxn = (int)1e5;
int v[maxn],cnt[maxn],N;
int res;
bool vis[maxn];
queue<int> Q;
void TopSort () {
while (!Q.empty()) {
Q.pop();
}
for (int i = 1; i <= N; i++) {
if (cnt[i] == 0) {
Q.push(i);
}
}
int now;
while (!Q.empty()) {
now = Q.front();
Q.pop();
vis[now] = true;
cnt[v[now]]--;
if (cnt[v[now]] == 0) {
Q.push(v[now]);
}
}
}
void DFS (int idx, int cnt) {
if (!vis[idx]) {
vis[idx] = true;
DFS(v[idx], cnt+1);
}else {
res = max(res, cnt);
}
}
int main() {
cin >> N;
memset(cnt, 0, sizeof(cnt));
memset(vis, false, sizeof(vis));
for (int i = 1; i <= N; i++) {
cin >> v[i];
cnt[v[i]]++; //入度++
}
TopSort(); //把不是环的check
res = 0;
for (int i = 1; i <= N; i++) {
if (!vis[i]) {
DFS(i, 0);
}
}
cout << res << endl;
return 0;
}