# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
503220 | _AnKahng | Traffic (IOI10_traffic) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int M = 1e6+5;
int n,p[M],id;
long long sum,f[M],res=LLONG_MAX;
vector<int> dsk[M];
void dfs(int u, int par){
long long mx = -1, o = p[u];
for(auto v : dsk[u]) if(v != par){
dfs(v,u);
mx = max(mx,f[v]);
o += f[v];
}
f[u] = o;
mx = max(mx,sum - o);
if(res > mx) res = mx, id = u;
}
int main()
{
cin >> n;
fo(i,1,n) cin >> p[i], sum = 0LL + sum + p[i];
fo(i,1,n-1){
int x,y;
cin >> x >> y;
x ++; y ++;
dsk[x].push_back(y);
dsk[y].push_back(x);
}
dfs(1,1);
cout << id - 1;
}