#include<bits/stdc++.h>
using namespace std;
// #pragma GCC optimize("trapv")
// #define int long long
const int N=1e6+10;
int n, vis[N], par[N], wpar[N], on_cycle[N];
vector<pair<int, int>> g[N];
long long pf[N], f[N][2];
void dfs(int u){
vis[u]=1;
long long mx1=-1e18, mx2=-1e18;
for (auto &e:g[u]){
int v=e.first, w=e.second;
if (on_cycle[v]) continue;
dfs(v);
long long val=f[v][0]+w;
f[u][0]=max(f[u][0], val);
f[u][1]=max(f[u][1], f[v][1]);
if (mx1<val) mx2=mx1, mx1=val;
else mx2=max(mx2, mx1);
}
f[u][1]=max({f[u][1], f[u][0], mx1+mx2});
}
int32_t main(){
ios_base::sync_with_stdio(false);
cin.tie(nullptr);
cin >> n;
for (int i=1; i<=n; ++i){
int j, k; cin >> j >> k;
g[j].emplace_back(i, k);
par[i]=j;
wpar[i]=k;
}
vector<long long> ans;
for (int i=1; i<=n; ++i) if (!vis[i]){
long long cur=0;
vector<int> cycle;
int u=i;
while (1){
if (vis[u]) break;
vis[u]=1;
cycle.push_back(u);
u=par[u];
}
cycle.erase(cycle.begin(), find(cycle.begin(), cycle.end(), u));
for (int v:cycle) on_cycle[v]=1;
for (int v:cycle){
dfs(v);
cur=max(cur, f[v][1]);
}
for (int j=1; j<=(int)cycle.size(); ++j){
pf[j]=pf[j-1]+wpar[cycle[j-1]];
}
long long sum=pf[(int)cycle.size()];
long long mx1=-1e18, mx2=-1e18;
for (int j=0; j<(int)cycle.size(); ++j){
cur=max(cur, f[cycle[j]][0]+pf[j]+mx1);
cur=max(cur, f[cycle[j]][0]-pf[j]+mx2);
mx1=max(mx1, f[cycle[j]][0]-pf[j]);
mx2=max(mx2, f[cycle[j]][0]+sum+pf[j]);
}
ans.push_back(cur);
}
cout << accumulate(ans.begin(), ans.end(), 0ll) << '\n';
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
24668 KB |
Output is correct |
2 |
Incorrect |
9 ms |
24668 KB |
Output isn't correct |
3 |
Correct |
9 ms |
24668 KB |
Output is correct |
4 |
Correct |
9 ms |
24668 KB |
Output is correct |
5 |
Correct |
9 ms |
24668 KB |
Output is correct |
6 |
Correct |
9 ms |
24668 KB |
Output is correct |
7 |
Correct |
9 ms |
24668 KB |
Output is correct |
8 |
Correct |
8 ms |
24668 KB |
Output is correct |
9 |
Correct |
8 ms |
24668 KB |
Output is correct |
10 |
Incorrect |
8 ms |
24668 KB |
Output isn't correct |
11 |
Correct |
9 ms |
24512 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
24668 KB |
Output is correct |
2 |
Correct |
10 ms |
24668 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
24668 KB |
Output is correct |
2 |
Correct |
9 ms |
24756 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
12 ms |
25384 KB |
Output is correct |
2 |
Correct |
15 ms |
26460 KB |
Output is correct |
3 |
Correct |
15 ms |
25844 KB |
Output is correct |
4 |
Incorrect |
10 ms |
25176 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
18 ms |
27484 KB |
Output is correct |
2 |
Correct |
24 ms |
30228 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
44 ms |
35040 KB |
Output is correct |
2 |
Correct |
45 ms |
39612 KB |
Output is correct |
3 |
Correct |
51 ms |
38960 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
73 ms |
46156 KB |
Output is correct |
2 |
Correct |
90 ms |
51140 KB |
Output is correct |
3 |
Correct |
96 ms |
65500 KB |
Output is correct |
4 |
Correct |
113 ms |
61376 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
137 ms |
59824 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
196 ms |
131072 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |