# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
373482 | eulerdesoja | Traffic (IOI10_traffic) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#include<fstream>
using namespace std;
#define int long long
#define pb push_back
#define sz(x) int(x.size())
typedef pair<int,int>ii;
typedef vector<int> vi;
void setIO(string s) {
ios_base::sync_with_stdio(0); cin.tie(0);
freopen((s+".in").c_str(),"r",stdin);
freopen((s+".out").c_str(),"w",stdout);
}
const int mxn=1e6+6;
vi g[mxn];
int n,a[mxn],ma[mxn],sum[mxn];//maximum of congestion in a road
void dfs(int i,int p){
for(int j:g[i])if(j!=p){
dfs(j,i);
sum[i]+=sum[j];
ma[i]=max(ma[i],max(ma[j],sum[j]));
}
sum[i]+=a[i];
}
int32_t main(){
ios_base::sync_with_stdio(0);cin.tie(0);
//setIO("sort");
cin>>n;
for(int i=0;i<n;i++)cin>>a[i];
for(int i=0;i<n-1;i++){
int x,y;cin>>x>>y;
g[x].pb(y);
g[y].pb(x);
}
int ans=1e18,res;
for(int i=0;i<n;i++){
memset(sum,0,sizeof(sum));
memset(ma,0,sizeof(ma));
dfs(i,-1);
if(ma[i]<ans){
ans=ma[i];
res=i;
}
}
cout<<res<<"\n";
return 0;
}
/*
5
10 10 10 20 20
0 2
1 2
2 3
3 4
*/