#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
const int maxn = 1e6+50;
const ll inf = 1e15+10;
int n;
int peso[maxn];
int firstCycle, lastCycle, backEdge;
int pai[maxn], edgePai[maxn];
ll pref[maxn];
ll maiorDist[maxn], maiorDist2;
int indDist;
bool mark[maxn], markCycle[maxn];
vector<pii> grafo[maxn];
void findCycle(int u, int p)
{
mark[u] = 1;
for (auto pp: grafo[u])
{
int v = pp.first, e = pp.second;
if (e == p) continue;
if (mark[v])
{
firstCycle = u, lastCycle = v;
backEdge = peso[e];
continue;
}
pai[v] = u, edgePai[v] = peso[e];
findCycle(v, e);
}
}
void dfs(int u, int p, int root, ll dist)
{
if (dist >= maiorDist[root])
indDist = u;
maiorDist[root] = max(maiorDist[root], dist);
for (auto pp: grafo[u])
{
int v = pp.first, e = pp.second;
if (v == p || markCycle[v]) continue;
dfs(v, u, root, dist+1ll*peso[e]);
}
}
void dfs2(int u, int p, ll dist)
{
maiorDist2 = max(maiorDist2, dist);
for (auto pp: grafo[u])
{
int v = pp.first, e = pp.second;
if (v == p || markCycle[v]) continue;
dfs2(v, u, dist+1ll*peso[e]);
}
}
int main(void)
{
scanf("%d", &n);
for (int i = 1; i <= n; i++)
{
int u, w;
scanf("%d %d", &u, &w);
peso[i] = w;
grafo[u].push_back({i, i});
grafo[i].push_back({u, i});
}
ll ans = 0;
vector<int> ciclo;
for (int T = 1; T <= n; T++)
{
if (mark[T]) continue;
findCycle(T, 0);
int at = lastCycle;
ll soma = backEdge;
ciclo.clear();
while (true)
{
ciclo.push_back(at);
markCycle[at] = 1;
if (at == firstCycle) break;
soma += 1ll*edgePai[at];
at = pai[at];
}
reverse(ciclo.begin(), ciclo.end());
ll M = 0, diam = 0;
for (int i = 0; i < ciclo.size(); i++)
{
int c = ciclo[i];
if (i > 0)
pref[i] = pref[i-1] + 1ll*edgePai[c];
dfs(c, 0, c, 0);
maiorDist2 = 0;
dfs2(indDist, 0, 0);
diam = max(diam, maiorDist2);
}
ll maxi1 = -inf, maxi2 = -inf;
for (int i = 0; i < ciclo.size(); i++)
{
int u = ciclo[i];
M = max(M, maiorDist[u]+pref[i] + maxi1);
maxi1 = max(maxi1, maiorDist[u]-pref[i]);
M = max(M, maiorDist[u]-pref[i] + maxi2 + soma);
maxi2 = max(maxi2, maiorDist[u]+pref[i]);
}
ans += max(M, diam);
}
printf("%lld\n", ans);
}
Compilation message
islands.cpp: In function 'int main()':
islands.cpp:120:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < ciclo.size(); i++)
~~^~~~~~~~~~~~~~
islands.cpp:137:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < ciclo.size(); i++)
~~^~~~~~~~~~~~~~
islands.cpp:78:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &n);
~~~~~^~~~~~~~~~
islands.cpp:83:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d", &u, &w);
~~~~~^~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
24 ms |
23800 KB |
Output is correct |
2 |
Correct |
28 ms |
23928 KB |
Output is correct |
3 |
Correct |
24 ms |
23932 KB |
Output is correct |
4 |
Correct |
26 ms |
23848 KB |
Output is correct |
5 |
Correct |
24 ms |
23828 KB |
Output is correct |
6 |
Correct |
23 ms |
23928 KB |
Output is correct |
7 |
Correct |
28 ms |
23800 KB |
Output is correct |
8 |
Correct |
24 ms |
23800 KB |
Output is correct |
9 |
Correct |
29 ms |
23804 KB |
Output is correct |
10 |
Incorrect |
24 ms |
23800 KB |
Output isn't correct |
11 |
Correct |
24 ms |
23920 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
25 ms |
23932 KB |
Output is correct |
2 |
Correct |
25 ms |
23928 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
25 ms |
24056 KB |
Output is correct |
2 |
Correct |
26 ms |
24184 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
33 ms |
25080 KB |
Output is correct |
2 |
Correct |
47 ms |
27276 KB |
Output is correct |
3 |
Correct |
39 ms |
25400 KB |
Output is correct |
4 |
Correct |
32 ms |
24696 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
61 ms |
28728 KB |
Output is correct |
2 |
Correct |
80 ms |
31992 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
132 ms |
39320 KB |
Output is correct |
2 |
Correct |
132 ms |
42252 KB |
Output is correct |
3 |
Correct |
162 ms |
51308 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
211 ms |
51796 KB |
Output is correct |
2 |
Correct |
265 ms |
69568 KB |
Output is correct |
3 |
Correct |
287 ms |
74476 KB |
Output is correct |
4 |
Correct |
365 ms |
88292 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
434 ms |
83416 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
618 ms |
131072 KB |
Output is correct |
2 |
Correct |
592 ms |
118788 KB |
Output is correct |
3 |
Runtime error |
555 ms |
131076 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
4 |
Halted |
0 ms |
0 KB |
- |