Submission #254239

#	Time	Username	Problem	Language	Result	Execution time	Memory
254239		tatyam	Traffic (IOI10_traffic)	C++17	100 / 100	1291 ms	129376 KiB

traffic

#include <bits/stdc++.h>
using namespace std;
template<class T, class U> bool chmax(T& a, const U& b){ if(a < b){ a = b; return 1; } return 0; }


int LocateCentre(int n, int p[], int s[], int d[]){
    const int sum = accumulate(p, p + n, 0);
    vector<vector<int>> g(n);
    for(int i = 0; i < n - 1; i++){
        const int a = s[i], b = d[i];
        g[a].push_back(b);
        g[b].push_back(a);
    }
    vector<int> dp(n);
    auto dfs = [&](int from, int at, const auto& dfs) -> void {
        for(int i : g[at]) if(i != from){
            dfs(at, i, dfs);
            p[at] += p[i];
            chmax(dp[at], p[i]);
        }
        chmax(dp[at], sum - p[at]);
    };
    dfs(-1, 0, dfs);
    return min_element(dp.begin(), dp.end()) - dp.begin();
}

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