Submission #554136

#	Time	Username	Problem	Language	Result	Execution time	Memory
554136		Alma	Traffic (IOI10_traffic)	C++17	50 / 100	132 ms	262144 KiB

traffic

#include <bits/stdc++.h>
#include "traffic.h"
using namespace std;
 
typedef long long int ll;
 
vector<vector<int>> graph;
vector<ll> cost;
vector<bool> visited;
vector<vector<ll>> DP;
 
void re_assign (int N, int idx) {
    visited.assign(N, false);
    visited[idx] = true;
}
 
ll traffic (int prevCity, int city) {
    if (DP[prevCity][city] != -1) return DP[prevCity][city];
 
    DP[prevCity][city] = cost[city];
    visited[city] = true;
 
    for (int road: graph[city]) {
        if (!visited[road]) DP[prevCity][city] += traffic(city, road);
    }
    return DP[prevCity][city];
}

int LocateCentre(int N, int pp[], int S[], int D[]) {
    graph.assign(N, vector<int> ());
    cost.clear();
    DP.assign(N, vector<ll> (N, -1));
 
    for (int i = 0; i < N; i++) {
        graph[S[i]].push_back(D[i]);
        graph[D[i]].push_back(S[i]);
        cost.push_back(pp[i]);
    }
    vector<pair<ll, int>> maximums (N);
    ll congestion = 0;
 
    for (int city = 0; city < N; city++) {
        ll maxCongestion = -1;
        re_assign(N, city);
 
        for (int road: graph[city]) {
            congestion = traffic (city, road);
            maxCongestion = max(maxCongestion, congestion);
        }   
        maximums[city].first = maxCongestion;
        maximums[city].second = city;
    }
    sort (maximums.begin(), maximums.end());
    pair<ll, int> arenaCity = *min_element(maximums.begin(), maximums.end());
    return arenaCity.second;
}

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