제출 #1347139

#제출 시각아이디문제언어결과실행 시간메모리
1347139nicolo_010꿈 (IOI13_dreaming)C++20
47 / 100
1099 ms108732 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;

void dfs(int u, int p=-1, int w=0) {
    h[u] = w;
    for (auto [v, pe] : adj[u]) {
        if (v==p) 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;
    vector<bool> vis(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;
        }
        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;
        }
        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;
        } 
        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<vector<pii>> vadj = adj;
        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});
        }
        h.assign(n, 0);
        dfs(0);
        int f = 0;
        for (int j=0; j<n; j++) {
            if (h[f] < h[j]) {
                f = j;
            }
        }
        h.assign(n, 0);
        dfs(f);
        int mx=0;
        for (int j=0; j<n; j++) {
            mx = max(mx, h[j]);
        }
        ans = min(ans, mx);
        adj = vadj;
    }
    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...