Submission #1347144

#TimeUsernameProblemLanguageResultExecution timeMemory
1347144nicolo_010Dreaming (IOI13_dreaming)C++20
47 / 100
1064 ms131072 KiB
#include <bits/stdc++.h>
#include "dreaming.h"
using namespace std;
using ll = long long;
using pii = pair<int, int>;

vector<int> h;
int n;
vector<vector<pii>> adj;
vector<bool> vis;

void dfs(int u, int p=-1, int w=0) {
    h[u] = w;
    vis[u] = true;
    for (auto [v, pe] : adj[u]) {
        if (v==p) continue;
        if (vis[v]) continue;
        dfs(v, u, w+pe);
    }
}

int t;
vector<int> comp;

void dfs1(int u, int p=-1) {
    comp.push_back(u);
    for (auto [v, w] : adj[u]) {
        if (v==p) continue;
        dfs1(v, u);
    }
}

int travelTime(int N, int m, int L, int A[], int B[], int T[]) {
    t=0;
    n = N;
    adj.assign(n, {});
    vector<vector<int>> cmp;
    vis.assign(n, false);
    for (int i=0; i<m; i++) {
        adj[A[i]].push_back({B[i], T[i]});
        adj[B[i]].push_back({A[i], T[i]});
    }
    for (int i=0; i<n; i++) {
        if (!vis[i]) {
            comp.clear();
            dfs1(i);
            cmp.push_back(comp);
            for (auto x : comp) {
                vis[x] = true;
            }
        }
    }
    int k = cmp.size();
    vector<int> l(k), r(k), best(k);
    h.assign(n, 0);
    vector<int> disl(n), disr(n);
    for (int i=0; i<k; i++) {
        for (auto v : cmp[i]) {
            h[v] = 0;
        }
        vis.assign(n, false);
        dfs(cmp[i][0]);
        l[i] = cmp[i][0];
        for (auto v : cmp[i]) {
            if (h[v] > h[l[i]]) {
                l[i] = v;
            }
        }
        for (auto v : cmp[i]) {
            h[v] = 0;
        }
        vis.assign(n, false);
        dfs(l[i]);
        r[i] = cmp[i][0];
        for (auto v : cmp[i]) {
            if (h[r[i]] < h[v]) {
                r[i] = v;
            }
        }
        for (auto v : cmp[i]) {
            disl[v] = h[v];
            h[v] = 0;
        }
        vis.assign(n, false);
        dfs(r[i]);
        for (auto v : cmp[i]) {
            disr[v] = h[v];
            h[v] = 0;
        }
        best[i] = cmp[i][0];
        for (auto v : cmp[i]) {
            int val = max(disl[best[i]], disr[best[i]]);
            //if (i==1) cout << best[i] << " " << val << " " << v << " " << max(disl[v], disr[v]) << endl;
            if (val > max(disl[v], disr[v])) {
                best[i] = v;
            }
        }
    }
    int ans = INT_MAX;
    for (int i=0; i<k; i++) {
        //elijo el k como root
        vector<int> callback(n, 0);
        for (int j=0; j<k; j++) {
            if (i==j) continue;
            adj[best[i]].push_back({best[j], L});
            adj[best[j]].push_back({best[i], L});
            callback[best[i]]++;
            callback[best[j]]++;
        }
        h.assign(n, 0);
        vis.assign(n, false);
        dfs(0);
        int f = 0;
        for (int j=0; j<n; j++) {
            if (h[f] < h[j]) {
                f = j;
            }
        }
        h.assign(n, 0);
        vis.assign(n, false);
        dfs(f);
        int mx=0;
        for (int j=0; j<n; j++) {
            mx = max(mx, h[j]);
        }
        ans = min(ans, mx);
        for (int i=0; i<n; i++) {
            for (int j=0; j<callback[i]; j++) {
                adj[i].pop_back();
            }
        }
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...