#include <bits/stdc++.h>
using namespace std;
#define N 1000001
vector<pair<int, long long> > adj[N];
bool in[N];
vector<int> s;
vector<int> cyc;
vector<long long> dp;
bool inCyc[N];
long long res;
long long dfs(int v, int p){
long long b1 = 0, b2 = 0;
for(auto xx : adj[v]){
int x = xx.first;
long long c = xx.second;
if(inCyc[x] || p == x) continue;
auto q = dfs(x, v);
long long up;
up = q + c;
if(b1 < up){
b2 = b1;
b1 = up;
} else b2 = max(b2, up);
}
res = max(res, b1 + b2);
return b1;
}
long long ans = 0;
unordered_map<int, unordered_map<int, long long> > lol;
vector<long long> p;
void solve(){
dp.clear();
p.clear();
cyc.clear();
res = 0;
for(int x : s){
if(inCyc[x]) {
p.push_back(p.empty() ? 0 : p.back() + lol[x][cyc.back()]);
cyc.push_back(x);
dp.push_back(dfs(x, 0));
}
}
long long full = p.back() + lol[cyc[0]][cyc.back()];
long long tmp = dp[0] - p[0];
for(int i = 1; i < cyc.size(); i++){
res = max(res, tmp + dp[i] + p[i]);
tmp = max(tmp, dp[i] - p[i]);
}
reverse(p.begin(), p.end());
for(auto& x : p){
x = full - x;
}
reverse(cyc.begin(), cyc.end());
reverse(dp.begin(), dp.end());
tmp = dp[0] - p[0];
for(int i = 1; i < cyc.size(); i++){
res = max(res, tmp + dp[i] + p[i]);
tmp = max(tmp, dp[i] - p[i]);
}
ans += res;
}
int trav(int i, int p){
s.push_back(i);
in[i] = 1;
int res = 0;
bool ok = 0;
for(auto xx : adj[i]){
int x = xx.first;
if(p == x && ok == 0){
ok = 1;
continue;
}
if(in[x]){
res = x;
continue;
}
int t = trav(x, i);
if(res < t)res = t;
}
if(res) inCyc[i] = 1;
if(res == i) return 0;
return res;
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n;
cin>>n;
for(int i = 1; i <= n; i++){
int e;
long long l;
cin>>e>>l;
adj[e].push_back({i, l});
adj[i].push_back({e, l});
lol[e][i] = max(lol[e][i], l);
lol[i][e] = max(lol[i][e], l);
}
for(int i = 1; i <= n; i++){
if(in[i]) continue;
s.clear();
trav(i, 0);
solve();
}
cout<<ans;
}
Compilation message
islands.cpp: In function 'void solve()':
islands.cpp:50:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 1; i < cyc.size(); i++){
~~^~~~~~~~~~~~
islands.cpp:62:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 1; i < cyc.size(); i++){
~~^~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
24 ms |
23808 KB |
Output is correct |
2 |
Incorrect |
23 ms |
23944 KB |
Output isn't correct |
3 |
Incorrect |
23 ms |
23936 KB |
Output isn't correct |
4 |
Correct |
23 ms |
23808 KB |
Output is correct |
5 |
Incorrect |
23 ms |
23768 KB |
Output isn't correct |
6 |
Correct |
24 ms |
23808 KB |
Output is correct |
7 |
Incorrect |
23 ms |
23800 KB |
Output isn't correct |
8 |
Incorrect |
23 ms |
23808 KB |
Output isn't correct |
9 |
Incorrect |
24 ms |
23904 KB |
Output isn't correct |
10 |
Correct |
23 ms |
23800 KB |
Output is correct |
11 |
Correct |
23 ms |
23936 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
24 ms |
24148 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
25 ms |
24312 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
51 ms |
27760 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
136 ms |
38532 KB |
Output is correct |
2 |
Incorrect |
150 ms |
50064 KB |
Output isn't correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
357 ms |
75268 KB |
Output is correct |
2 |
Correct |
379 ms |
86392 KB |
Output is correct |
3 |
Incorrect |
447 ms |
103172 KB |
Output isn't correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
541 ms |
121876 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
431 ms |
131072 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Runtime error |
451 ms |
131072 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
2 |
Halted |
0 ms |
0 KB |
- |