| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 372416 | MaisyDoge13 | Traffic (IOI10_traffic) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <cstdio>
#include <vector>
#include <utility>
#include <cmath>
#include <algorithm>
#include <array>
#include <set>
#include <climits>
#include <map>
#include <memory>
using namespace std;
#define input "problem.in"
#define output "problem.out"
int n;
struct City {
int i, p;
bool vis=0;
vector<pair<int,shared_ptr<City>>> roads;
City(int I, int P) {
i=I; p=P;
}
int DFS(int prev=-1) {
int added=p;
for (int j=0;j<(roads.size());j++) {
if (roads[j].second->i==prev) continue;
if (roads[j].first==0) {
roads[j].first=roads[j].second->DFS(i);
}
added+=roads[j].first;
}
return added;
}
int res() {
int res=0;
for (int j=0;j<(roads.size());j++) {
res=max(roads[j].first, res);
}
return res;
}
};
vector<shared_ptr<City>> cities;
int main() {
ios_base::sync_with_stdio(0); cin.tie(0);
//freopen(input, "r", stdin);
//freopen(output, "w", stdout);
cin >> n; cities.reserve(n);
for (int i=0;i<n;i++) {
int p; cin >> p;
cities.push_back(make_shared<City>(i, p));
}
for (int i=1;i<n;i++) {
int a, b;
cin >> a >> b;
cities[a]->roads.push_back({0,cities[b]});
cities[b]->roads.push_back({0,cities[a]});
}
int ans=INT_MAX;
int city=0;
for (shared_ptr<City> c: cities) {
c->DFS();
}
for (shared_ptr<City> c: cities) {
int res=c->res();
if (res < ans) {
ans = res;
city=c->i;
}
}
cout << city << endl;
}