oj.uz

Submission #255317

# Submission time Handle Problem Language Result Execution time Memory
255317 2020-07-31T17:27:10 Z rama_pang Beads and wires (APIO14_beads) C++14
100 / 100
 641 ms 51800 KB 
#include <bits/stdc++.h>
using namespace std;

using lint = long long;
const int MAXN = 2e5 + 5;
const lint INF = 1e18;

int N;
vector<pair<int, int>> adj[MAXN];

lint dp[MAXN];
multiset<lint> mx[MAXN];

long long ans = 0;

void Dfs(int u, int p = -1) {
  dp[u] = 0;
  mx[u].emplace(-INF);
  for (auto e : adj[u]) if (e.first != p) {
    int v = e.first;
    int w = e.second;
    Dfs(v, u);
    dp[u] += max(dp[v], w + dp[v] + *prev(end(mx[v])));
    mx[u].emplace(dp[v] + w - max(dp[v], w + dp[v] + *prev(end(mx[v]))));
  }
}

void Reroot(int u, int p = -1) {
  ans = max(ans, dp[u]);
  for (auto e : adj[u]) if (e.first != p) {
    int v = e.first;
    int w = e.second;

    dp[u] -= max(dp[v], w + dp[v] + *prev(end(mx[v])));
    mx[u].erase(mx[u].find(dp[v] + w - max(dp[v], w + dp[v] + *prev(end(mx[v])))));

    dp[v] += max(dp[u], w + dp[u] + *prev(end(mx[u])));
    mx[v].emplace(dp[u] + w - max(dp[u], w + dp[u] + *prev(end(mx[u]))));

    Reroot(v, u);

    dp[v] -= max(dp[u], w + dp[u] + *prev(end(mx[u])));
    mx[v].erase(mx[v].find(dp[u] + w - max(dp[u], w + dp[u] + *prev(end(mx[u])))));

    dp[u] += max(dp[v], w + dp[v] + *prev(end(mx[v])));
    mx[u].emplace(dp[v] + w - max(dp[v], w + dp[v] + *prev(end(mx[v]))));
  }
}

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0), cout.tie(0);

  cin >> N;
  for (int i = 0; i + 1 < N; i++) {
    int u, v, w;
    cin >> u >> v >> w;
    u--, v--;
    adj[u].emplace_back(v, w);
    adj[v].emplace_back(u, w);
  }
  Dfs(0), Reroot(0);
  cout << ans << "\n";
  return 0;
}

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 14464 KB Output is correct
2 Correct 9 ms 14464 KB Output is correct
3 Correct 9 ms 14464 KB Output is correct
4 Correct 9 ms 14464 KB Output is correct
5 Correct 11 ms 14464 KB Output is correct
6 Correct 10 ms 14464 KB Output is correct
7 Correct 9 ms 14464 KB Output is correct
8 Correct 11 ms 14464 KB Output is correct
9 Correct 10 ms 14464 KB Output is correct
10 Correct 10 ms 14460 KB Output is correct
11 Correct 9 ms 14464 KB Output is correct
12 Correct 9 ms 14372 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 14464 KB Output is correct
2 Correct 9 ms 14464 KB Output is correct
3 Correct 9 ms 14464 KB Output is correct
4 Correct 9 ms 14464 KB Output is correct
5 Correct 11 ms 14464 KB Output is correct
6 Correct 10 ms 14464 KB Output is correct
7 Correct 9 ms 14464 KB Output is correct
8 Correct 11 ms 14464 KB Output is correct
9 Correct 10 ms 14464 KB Output is correct
10 Correct 10 ms 14460 KB Output is correct
11 Correct 9 ms 14464 KB Output is correct
12 Correct 9 ms 14372 KB Output is correct
13 Correct 11 ms 14464 KB Output is correct
14 Correct 9 ms 14464 KB Output is correct
15 Correct 11 ms 14464 KB Output is correct
16 Correct 11 ms 14464 KB Output is correct
17 Correct 9 ms 14464 KB Output is correct
18 Correct 11 ms 14464 KB Output is correct
19 Correct 9 ms 14464 KB Output is correct
20 Correct 9 ms 14464 KB Output is correct
21 Correct 9 ms 14464 KB Output is correct
22 Correct 15 ms 14464 KB Output is correct

Subtask #3
29.0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 14464 KB Output is correct
2 Correct 9 ms 14464 KB Output is correct
3 Correct 9 ms 14464 KB Output is correct
4 Correct 9 ms 14464 KB Output is correct
5 Correct 11 ms 14464 KB Output is correct
6 Correct 10 ms 14464 KB Output is correct
7 Correct 9 ms 14464 KB Output is correct
8 Correct 11 ms 14464 KB Output is correct
9 Correct 10 ms 14464 KB Output is correct
10 Correct 10 ms 14460 KB Output is correct
11 Correct 9 ms 14464 KB Output is correct
12 Correct 9 ms 14372 KB Output is correct
13 Correct 11 ms 14464 KB Output is correct
14 Correct 9 ms 14464 KB Output is correct
15 Correct 11 ms 14464 KB Output is correct
16 Correct 11 ms 14464 KB Output is correct
17 Correct 9 ms 14464 KB Output is correct
18 Correct 11 ms 14464 KB Output is correct
19 Correct 9 ms 14464 KB Output is correct
20 Correct 9 ms 14464 KB Output is correct
21 Correct 9 ms 14464 KB Output is correct
22 Correct 15 ms 14464 KB Output is correct
23 Correct 13 ms 15232 KB Output is correct
24 Correct 13 ms 15232 KB Output is correct
25 Correct 13 ms 15232 KB Output is correct
26 Correct 18 ms 16000 KB Output is correct
27 Correct 18 ms 16000 KB Output is correct
28 Correct 20 ms 16000 KB Output is correct
29 Correct 20 ms 16000 KB Output is correct
30 Correct 20 ms 15928 KB Output is correct
31 Correct 19 ms 16512 KB Output is correct

Subtask #4
43.0 / 43.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 14464 KB Output is correct
2 Correct 9 ms 14464 KB Output is correct
3 Correct 9 ms 14464 KB Output is correct
4 Correct 9 ms 14464 KB Output is correct
5 Correct 11 ms 14464 KB Output is correct
6 Correct 10 ms 14464 KB Output is correct
7 Correct 9 ms 14464 KB Output is correct
8 Correct 11 ms 14464 KB Output is correct
9 Correct 10 ms 14464 KB Output is correct
10 Correct 10 ms 14460 KB Output is correct
11 Correct 9 ms 14464 KB Output is correct
12 Correct 9 ms 14372 KB Output is correct
13 Correct 11 ms 14464 KB Output is correct
14 Correct 9 ms 14464 KB Output is correct
15 Correct 11 ms 14464 KB Output is correct
16 Correct 11 ms 14464 KB Output is correct
17 Correct 9 ms 14464 KB Output is correct
18 Correct 11 ms 14464 KB Output is correct
19 Correct 9 ms 14464 KB Output is correct
20 Correct 9 ms 14464 KB Output is correct
21 Correct 9 ms 14464 KB Output is correct
22 Correct 15 ms 14464 KB Output is correct
23 Correct 13 ms 15232 KB Output is correct
24 Correct 13 ms 15232 KB Output is correct
25 Correct 13 ms 15232 KB Output is correct
26 Correct 18 ms 16000 KB Output is correct
27 Correct 18 ms 16000 KB Output is correct
28 Correct 20 ms 16000 KB Output is correct
29 Correct 20 ms 16000 KB Output is correct
30 Correct 20 ms 15928 KB Output is correct
31 Correct 19 ms 16512 KB Output is correct
32 Correct 69 ms 22264 KB Output is correct
33 Correct 81 ms 22136 KB Output is correct
34 Correct 67 ms 22136 KB Output is correct
35 Correct 461 ms 45944 KB Output is correct
36 Correct 481 ms 45944 KB Output is correct
37 Correct 394 ms 45816 KB Output is correct
38 Correct 518 ms 46200 KB Output is correct
39 Correct 641 ms 46168 KB Output is correct
40 Correct 591 ms 46176 KB Output is correct
41 Correct 485 ms 51800 KB Output is correct