제출 #796878

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
796878		Liudas	Traffic (IOI10_traffic)	C++17	100 / 100	1852 ms	172464 KiB

traffic

#include <bits/stdc++.h>
#include "traffic.h"
using namespace std;
int LocateCentre(int N, int P[], int S[], int D[]){
    vector<set<int>> tree(N);
    vector<int> maxim(N), total(N);
    priority_queue<pair<int, int>> que;
    for(int i = 0; i < N-1; i ++){
        tree[S[i]].insert(D[i]);
        tree[D[i]].insert(S[i]);
    }
    for(int i = 0; i < N; i ++){
        if(tree[i].size() == 1){
            que.push({-P[i], i});
        }
    }
    while(!que.empty()){
        int val = -que.top().first, a = que.top().second;
        que.pop();
        for(int i : tree[a]){
            tree[i].erase(a);
            maxim[i] = max(val, maxim[i]);
            total[i] += val;
            if(tree[i].size() == 1){
                que.push({-(total[i]+P[i]), i});
            }
            if(tree[i].size() == 0){
                return i;
            }
        }
    }
    return 0;
}

• Subtask 1: 25 / 25
• Subtask 2: 25 / 25
• Subtask 3: 25 / 25
• Subtask 4: 25 / 25