# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
554173 | Alma | Traffic (IOI10_traffic) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
// #include "traffic.h"
using namespace std;
typedef long long int ll;
vector<vector<int>> graph;
vector<ll> cost;
vector<bool> visited;
vector<ll> DP;
int arenaCity;
ll traffic (int city) {
if (DP[city] != -1) return DP[city];
DP[city] = cost[city];
visited[city] = true;
for (int road: graph[city]) {
if (!visited[road]) DP[city] += traffic(road);
}
return DP[city];
}
void untraffic (int city, int cityMaxDiff) {
int bestCity = city;
ll bestDiff = cityMaxDiff, maxCongestion = DP[city];
for (int road: graph[city]) {
if (DP[road] < maxCongestion) continue;
ll diff = maxCongestion - DP[road];
if (diff < bestDiff) {
bestCity = road;
bestDiff = diff;
}
}
if (bestCity != city) {
DP[city] -= DP[bestCity];
DP[bestCity] = maxCongestion;
arenaCity = bestCity;
untraffic(bestCity, DP[bestCity] - DP[city])
}
}
int LocateCentre(int N, int pp[], int S[], int D[]) {
graph.assign(N, vector<int>());
cost.clear();
DP.assign(N, -1);
for (int i = 0; i < N; i++) {
graph[S[i]].push_back(D[i]);
graph[D[i]].push_back(S[i]);
cost.push_back(pp[i]);
}
visited.assign(N, false);
ll congestion = traffic(0);
congestion = 0;
for (auto road: graph[0]) {
congestion = max(congestion, DP[0] - DP[road]);
}
arenaCity = 0;
untraffic(0, congestion);
return arenaCity;
}