제출 #796887

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

traffic

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

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