# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
766408 | irmuun | Traffic (IOI10_traffic) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#include "traffic.h"
using namespace std;
#define pb push_back
#define ll long long
#define ff first
#define ss second
#define all(s) s.begin(),s.end()
const int maxn=1e6;
const ll INF=1e18;
vector<int>adj[maxn],used(maxn,0),par(n);
ll tot[maxn];
int LocateCentre(int N, int P[], int S[], int D[]) {
ll all=0;
for(int i=0;i<N-1;i++){
adj[S[i]].pb(D[i]);
adj[D[i]].pb(S[i]);
}
for(int i=0;i<N;i++){
all+=P[i];
}
function <void(int)> dfs=[&](int x){
used[x]=1;
tot[x]+=P[x];
for(auto y:adj[x]){
if(used[y]==0){
par[y]=x;
dfs(y);
tot[x]+=P[y];
}
}
};
function <void(int,int)> num=[&](int x,int y){
if(par[y]==x){
return tot[y];
}
else{
return all-tot[x];
}
};
dfs(0);
p[0]=-1;
ll mn=INF,ans=0;
for(int i=0;i<n;i++){
ll curmin=INF;
for(auto y:adj[i]){
curmin=min(curmin,num(i,y));
}
if(curmin<mn){
mn=curmin;
ans=i;
}
}
return ans;
}