Submission #1073968

#	Time	Username	Problem	Language	Result	Execution time	Memory
1073968		Ignut	Dreaming (IOI13_dreaming)	C++17	10 / 100	1053 ms	9948 KiB

dreaming

// Ignut

#include <bits/stdc++.h>
#include "dreaming.h"

using namespace std;
using ll = long long;

const int INF = 1e9 + 123;
const int MAXN = 1e5 + 123;

vector<pair<int, int>> tree[MAXN];
int dist[MAXN];

void dfs(int v, int par, int d) {
    dist[v] = d;
    for (auto [to, w] : tree[v])
        if (to != par)
            dfs(to, v, d + w); 
}

void FindDist(int source) {
    dfs(source, -1, 0);
}

bool used[MAXN];
vector<int> comp;

void dfs0(int v, int par = -1) {
    used[v] = true;
    comp.push_back(v);
    for (auto [to, w] : tree[v])
        if (to != par)
            dfs0(to, v);
}

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
    for (int i = 0; i < N; i ++) {
        tree[i].clear();
        used[i] = false;
    }
    for (int i = 0; i < M; i ++) {
        tree[A[i]].push_back({B[i], T[i]});
        tree[B[i]].push_back({A[i], T[i]});
    }
    
    vector<int> c;
    for (int i = 0; i < N; i ++) {
        if (used[i]) continue;
        comp.clear();
        dfs0(i);
        int ctr = 0, maxD = INF;
        for (int v : comp) {
            FindDist(v);
            int md = 0;
            for (int u : comp) md = max(md, dist[u]);
            if (md < maxD) {
                ctr = v;
                maxD = md;
            }
        }
        c.push_back(ctr);
    }

    if (c.size() == 2) {
        tree[c[0]].push_back({c[1], L});
        tree[c[1]].push_back({c[0], L});
    }

    int res = 0;
    for (int i = 0; i < N; i ++) {
        FindDist(i);
        for (int j = 0; j < N; j ++) res = max(res, dist[j]);
    }
    return res;
}

• Subtask 1: 0 / 14
• Subtask 2: 10 / 10
• Subtask 3: 0 / 23
• Subtask 4: 0 / 18
• Subtask 5: 0 / 12
• Subtask 6: 0 / 23