#include <iostream>
#include <queue>
#include <cassert>
#include <set>
#include <vector>
#include <algorithm>
using namespace std;
typedef pair<int, int> P;
#define rep(i, n) for (int i=0; i<(n); i++)
#define all(x) x.begin(), x.end()
#define uniq(x) x.erase(unique(all(x)), x.end())
#define index(x, y) (int)(lower_bound(all(x), y) - x.begin())
#define pb push_back
#define _1 first
#define _2 second
#define INF (1LL<<60)
int N;
long long K;
int A[100000];
int F[60][100000];
long long C[100000];
long long dist[3000][3000];
int dst[3000][3000];
inline int f(int x, long long k) {
for (int i=59; i>=0; i--) if ((k>>i)&1) x = F[i][x];
return x;
}
signed main() {
cin >> N >> K;
if (N > 3000) abort();
rep(i, N) cin >> A[i], A[i]--, F[0][i] = A[i];
rep(i, 59) rep(j, N) F[i+1][j] = F[i][F[i][j]];
rep(i, N) {
rep(j, N) dist[i][j] = INF;
int x = i;
rep(_, N) {
dist[i][x] = min(dist[i][x], (long long)_);
dst[i][_] = x;
x = A[x];
}
}
int def_dst = f(0, K);
rep(a, N) {
long long d = dist[0][a];
if (K <= d) {
C[def_dst] += N;
continue;
}
long long K2 = K-d;
rep(b, N) {
if (dist[b][a] == INF) {
C[f(b, K2-1)]++;
}
else {
int len = 1+dist[b][a];
long long s = K2%len;
if (s == 0) C[a]++;
else C[dst[b][s-1]]++;
}
}
}
rep(i, N) cout << C[i] << "\n";
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
132096 KB |
Output is correct |
2 |
Correct |
0 ms |
132096 KB |
Output is correct |
3 |
Correct |
1 ms |
132096 KB |
Output is correct |
4 |
Correct |
2 ms |
132096 KB |
Output is correct |
5 |
Correct |
0 ms |
132096 KB |
Output is correct |
6 |
Correct |
1 ms |
132096 KB |
Output is correct |
7 |
Correct |
1 ms |
132096 KB |
Output is correct |
8 |
Correct |
0 ms |
132096 KB |
Output is correct |
9 |
Correct |
1 ms |
132096 KB |
Output is correct |
10 |
Correct |
0 ms |
132096 KB |
Output is correct |
11 |
Correct |
1 ms |
132096 KB |
Output is correct |
12 |
Correct |
2 ms |
132096 KB |
Output is correct |
13 |
Correct |
0 ms |
132096 KB |
Output is correct |
14 |
Correct |
1 ms |
132096 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
103 ms |
132096 KB |
Output is correct |
2 |
Correct |
33 ms |
132096 KB |
Output is correct |
3 |
Correct |
123 ms |
132096 KB |
Output is correct |
4 |
Execution timed out |
2000 ms |
132096 KB |
Execution timed out |
5 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
0 ms |
132096 KB |
Execution killed because of forbidden syscall gettid (186) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
0 ms |
132096 KB |
Execution killed because of forbidden syscall gettid (186) |
2 |
Halted |
0 ms |
0 KB |
- |