Submission #427759

#	Time	Username	Problem	Language	Result	Execution time	Memory
427759		haxorman	Traffic (IOI10_traffic)	C++14	50 / 100	484 ms	156900 KiB

traffic

#include "traffic.h"
#include "bits/stdc++.h"
using namespace std;

const int mxN = 1e6 + 7;

int n, p[mxN];
long long dp[mxN], tot;
vector<int> graph[mxN];

void dfs(int u, int par) {
   dp[u] = p[u];

   for (auto v : graph[u]) {
      if (v != par) {
         dfs(v, u);
         dp[u] += dp[v];
      }
   }
}

int LocateCentre(int N, int P[], int S[], int D[]) {
   n = N;

   for (int i = 0; i < n; ++i) {
      p[i] = P[i];
      tot += p[i];
   }

   for (int i = 0; i < n - 1; ++i) {
      graph[S[i]].push_back(D[i]);
      graph[D[i]].push_back(S[i]);
   }

   dfs(0, -1);

   long long res = INT_MAX, city = 0;
   for (int u = 0; u < n; ++u) {
      if (res > max(tot - dp[u], dp[u] - p[u])) {
         res = max(tot - dp[u], dp[u] - p[u]);
         city = u;
      }
   }
   return city;
}

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