Submission #296370

#	Time	Username	Problem	Language	Result	Execution time	Memory
296370		morato	Traffic (IOI10_traffic)	C++17	0 / 100	17 ms	23808 KiB

traffic

#include <bits/stdc++.h>
using namespace std;

const int maxn = 1e6 + 5;

vector<int> adj[maxn];
int p[maxn], h[maxn], sub[maxn], tot;

void dfs(int v, int _p = -1) {
  for (int u : adj[v]) if (u != _p) {
    dfs(u, v);
    sub[v] += sub[u] + p[u];
    h[v] = max(h[v], sub[u] + p[u]);
  }
}

int LocateCentre(int N, int *P, int *S, int *D) {
  for (int i = 0; i < N; i++) {
    p[i] = P[i];
    tot += p[i];
  }
  for (int i = 0; i < N; i++) {
    adj[S[i]].push_back(D[i]);
    adj[D[i]].push_back(S[i]);
  }
  dfs(0);
  int ans = 0, heavy = h[0];
  for (int i = 1; i < N; i++) {
    h[i] = max(h[i], tot - sub[i] - p[i]);
    if (h[i] < heavy) {
      heavy = h[i];
      ans = i;
    }
  }
  return ans;
}

• Subtask 1: 0 / 25
• Subtask 2: 0 / 25
• Subtask 3: 0 / 25
• Subtask 4: 0 / 25