Submission #1073983

#	Time	Username	Problem	Language	Result	Execution time	Memory
1073983		Ignut	Dreaming (IOI13_dreaming)	C++17	0 / 100	1067 ms	11732 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);
        FindDist(i);
        // map<int, int> d1;
        // for (int v : comp) d1[v] = dist[v];
        vector<int> d1;
        for (int i = 0; i < N; i ++) d1.push_back(dist[i]);

        int j = i;
        for (int v : comp) if (dist[v] > dist[j]) j = v;
        FindDist(j);

        int ctr = i, maxDist = INF;
        for (int v : comp) {
            FindDist(j);
            int d = max(d1[v], dist[v]);
            int dd = 0;
            FindDist(v);
            for (int u : comp) dd = max(dd, dist[u]);
            if (d != dd) {
                N /= 0;
            }
            if (d < maxDist) {
                ctr = v;
                maxDist = d;
            }
        }
        c.push_back(ctr);
        // cout << "-- " << ctr << '\n';
    }

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

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

Compilation message (stderr)

dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:70:19: warning: division by zero [-Wdiv-by-zero]
   70 |                 N /= 0;
      |                 ~~^~~~

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