oj.uz

Submission #875547

# Submission time Handle Problem Language Result Execution time Memory
875547 2023-11-20T03:27:40 Z NeroZein Fireworks (APIO16_fireworks) C++17
0 / 100
 0 ms 348 KB 
#include "bits/stdc++.h"
using namespace std;

#ifdef Nero
#include "Deb.h"
#else
#define deb(...)
#endif

const int N = 5002;

bool h[N]; 
long long dis[N]; 
long long mn[N], mx[N]; 
vector<pair<int, int>> g[N]; 

void dfs(int v) {
  if (g[v].size() == 0) {
    mn[v] = mx[v] = dis[v];
    return; 
  }
  mn[v] = LLONG_MAX;
  mx[v] = -LLONG_MAX;
  for (auto [u, w] : g[v]) {
    dis[u] = dis[v] + w; 
    dfs(u); 
    mn[v] = min(mn[v], mn[u]);
    mx[v] = max(mx[v], mx[u]);
  }
}

long long dfs2(int v, long long add, const long long req) {
  long long ret = 0;
  for (auto [u, w] : g[v]) {
    long long t = 0, r = 1; 
    if (mn[u] + add > req) {
      assert(add <= 0); 
      r = -1; 
      t = min((long long) w, mn[u] + add - req); 
    } else if (mx[u] + add < req) {
      assert(add >= 0);
      t = min((long long) w, req - (mx[u] + add)); 
    }
    ret += t + dfs2(u, add + t * r, req); 
  }
  return ret;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int n, m;
  cin >> n >> m;
  vector<int> cs;
  for(int i = 2; i <= n + m; ++i) {
    int p, c;
    cin >> p >> c;
    g[p].emplace_back(i, c); 
    if (i > n) {
      cs.push_back(c); 
    }
    assert(p <= n); 
  }
  sort(cs.begin(), cs.end()); 
  dfs(1); 
  long long ans = LLONG_MAX; 
  //for (int i = 1; i <= 605; ++i) {
  long long req = cs[cs.size() / 2]; 
    ans = min(ans, dfs2(1, 0, req)); 
  //}
  cout << ans << '\n'; 
  return 0;
}

Subtask #1
0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 19.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 29.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 0 ms 348 KB Output isn't correct
3 Halted 0 ms 0 KB -