Submission #1305039

#TimeUsernameProblemLanguageResultExecution timeMemory
1305039flyDreaming (IOI13_dreaming)C++20
0 / 100
29 ms12008 KiB
#include <bits/stdc++.h>
#include "dreaming.h"
using namespace std;
vector<vector<pair<int, int>>> connections(100000);
vector<bool> visited(100000);
vector<pair<int, int>> far(100000, {-1, 0});
vector<pair<int, int>> sfar(100000, {-1, 0});
int treeans;
vector<int> answers;
void fdfs(int cur, int par) {
    visited[cur] = true;
    for (pair<int, int> next: connections[cur]) {
        if (next.first == par) continue;
        fdfs(next.first, cur);
        int d = far[next.first].second + next.second;
        if (d > far[cur].second) {
            sfar[cur] = far[cur];
            far[cur].first = next.first;
            far[cur].second = d;
        } else if (d > sfar[cur].second) {
            sfar[cur].first = next.first;
            sfar[cur].second = d;
        }
    }
}
void sdfs(int cur, int par) {
    for (pair<int, int> next: connections[cur]) {
        if (next.first == par) continue;
        if (far[cur].first != next.first) {
            if (far[cur].second + next.second > far[next.first].second) {
                sfar[next.first] = far[next.first];
                far[next.first] = {cur, far[cur].second + next.second};
            } else if (far[cur].second + next.second > sfar[next.first].second) {
                sfar[next.first] = {cur, far[cur].second + next.second};
            }
        } else {
            if (sfar[cur].second + next.second > far[next.first].second) {
                sfar[next.first] = far[next.first];
                far[next.first] = {cur, sfar[cur].second + next.second};
            } else if (far[cur].second + next.second > sfar[next.first].second) {
                sfar[next.first] = {cur, sfar[cur].second + next.second};
            }
        }
        sdfs(next.first, cur);
    }
    treeans = min(treeans, far[cur].second);
}
int travelTime(int n, int m, int l, int A[], int B[], int T[]) {
    for (int i = 0; i < m; ++i) {
        connections[A[i]].push_back({B[i], T[i]});
        connections[B[i]].push_back({A[i], T[i]});
    }
    for (int i = 0; i < n; ++i) {
        if (!visited[i]) {
            fdfs(i, -1);
            treeans = INT_MAX;
            sdfs(i, -1);
            answers.push_back(treeans);
        }
    }
    sort(answers.begin(), answers.end(), greater<int>());
    if (answers.size() > 1) {
        return (answers[0] + answers[1] + l);
    } else {
        return answers[0];
    }
}
#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...