#pragma GCC optimize ("Ofast")
#pragma GCC target ("avx2")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int MOD = 1e9+7;
const int MX = 1e6+10;
const ll INF = 0x3f3f3f3f3f3f3f3f;
int n, m, a[MX], w[MX];
vector<int> adj[MX];
ll ans, ret;
vector<ll> V, S;
bool vv[MX];
inline int cycle(int u) {
if(vv[u]) return u;
vv[u] = 1;
int r = cycle(a[u]);
if(r) {
V.push_back(u);
S.push_back(w[u]);
if(r==u) return 0;
else return r;
}
vv[u] = 0;
return 0;
}
inline ll dfs(int u) {
ll mx = 0;
vv[u] = 1;
for(int v : adj[u]) {
if(!vv[v]) {
ll r = dfs(v) + w[v];
ret = max(ret, mx+r);
mx = max(mx, r);
}
}
return mx;
}
void solve(int u) {
V.clear(), S.clear();
cycle(u);
reverse(V.begin(), V.end());
reverse(S.begin(), S.end());
for(auto &v : V) v = dfs(v);
m = V.size();
for(int i = m-1; i > 0; --i) swap(S[i], S[i-1]);
for(int i = 1; i < m; ++i) S[i] += S[i-1];
ll m1 = -INF, m2 = -INF;
for(int i = 0; i < m; ++i) {
ret = max(ret, max(m1+S[i]+V[i], m2-S[i]+V[i]+S[m-1]));
m1 = max(m1, V[i]-S[i]);
m2 = max(m2, V[i]+S[i]);
}
}
int main() {
cin.tie(0)->sync_with_stdio(0);
cin >> n;
for(int i = 1; i <= n; ++i) {
cin >> a[i] >> w[i];
adj[a[i]].push_back(i);
}
for(int i = 1; i <= n; ++i)
if(!vv[i]) solve(i), ans+=ret, ret = 0;
cout << ans << "\n";
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
19 ms |
23788 KB |
Output is correct |
2 |
Correct |
17 ms |
23788 KB |
Output is correct |
3 |
Correct |
17 ms |
23916 KB |
Output is correct |
4 |
Correct |
17 ms |
23788 KB |
Output is correct |
5 |
Correct |
18 ms |
23788 KB |
Output is correct |
6 |
Correct |
18 ms |
23788 KB |
Output is correct |
7 |
Correct |
17 ms |
23788 KB |
Output is correct |
8 |
Correct |
18 ms |
23788 KB |
Output is correct |
9 |
Correct |
17 ms |
23788 KB |
Output is correct |
10 |
Correct |
17 ms |
23788 KB |
Output is correct |
11 |
Correct |
18 ms |
23788 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
17 ms |
23916 KB |
Output is correct |
2 |
Correct |
17 ms |
23916 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
17 ms |
23916 KB |
Output is correct |
2 |
Correct |
18 ms |
24172 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
22 ms |
24684 KB |
Output is correct |
2 |
Correct |
32 ms |
26348 KB |
Output is correct |
3 |
Correct |
26 ms |
24812 KB |
Output is correct |
4 |
Correct |
21 ms |
24300 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
36 ms |
27240 KB |
Output is correct |
2 |
Correct |
55 ms |
28908 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
31596 KB |
Output is correct |
2 |
Correct |
88 ms |
39432 KB |
Output is correct |
3 |
Correct |
107 ms |
44040 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
127 ms |
37996 KB |
Output is correct |
2 |
Correct |
167 ms |
52688 KB |
Output is correct |
3 |
Correct |
187 ms |
68816 KB |
Output is correct |
4 |
Correct |
233 ms |
73504 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
241 ms |
37596 KB |
Output is correct |
2 |
Correct |
548 ms |
89160 KB |
Output is correct |
3 |
Correct |
234 ms |
48236 KB |
Output is correct |
4 |
Correct |
313 ms |
76368 KB |
Output is correct |
5 |
Correct |
309 ms |
74240 KB |
Output is correct |
6 |
Correct |
722 ms |
52588 KB |
Output is correct |
7 |
Correct |
361 ms |
101576 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
376 ms |
128364 KB |
Output is correct |
2 |
Correct |
336 ms |
95980 KB |
Output is correct |
3 |
Runtime error |
350 ms |
131076 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
4 |
Halted |
0 ms |
0 KB |
- |