Submission #482960

#TimeUsernameProblemLanguageResultExecution timeMemory
482960jalsolCrocodile's Underground City (IOI11_crocodile)C++17
100 / 100
534 ms72468 KiB
#include "crocodile.h"

#ifdef LOCAL
#include "local.h"
#else
#include <bits/stdc++.h>
#define debug(...)
#define DB(...)
#endif

using namespace std;

const bool __initialization = []() {
    cin.tie(nullptr)->sync_with_stdio(false);
    cout << setprecision(12) << fixed;
    return true;
}();

using ll = long long;
using ld = long double;

#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()

#define For(i, l, r) for (int i = (l); i <= (r); ++i)
#define Ford(i, r, l) for (int i = (r); i >= (l); --i)
#define Rep(i, n) For (i, 0, (n) - 1)
#define Repd(i, n) Ford (i, (n) - 1, 0)
#define Fe(...) for (auto __VA_ARGS__)

template <class C> inline int isz(const C& c) { return static_cast<int>(c.size()); }
template <class T> inline bool chmin(T& a, const T& b) { return (a > b) ? a = b, true : false; }
template <class T> inline bool chmax(T& a, const T& b) { return (a < b) ? a = b, true : false; }

constexpr ld eps = 1e-9;
constexpr int inf = 1e9;
constexpr ll linf = 1e18;

// =============================================================================

constexpr int maxn = 1e5 + 5;

struct Link { int v, c; };

struct Data {
    int u, du;
    bool operator<(const Data& o) const {
        return du > o.du;
    }
};

int n, ne, nexit;
vector<Link> g[maxn];
int esc[maxn];

int cnt[maxn];
priority_queue<Data> q;

int Dijkstra() {
    while (isz(q)) {
        auto [u, du] = q.top(); q.pop();

        if (++cnt[u] == 2) {
            if (u == 0) return du;

            Fe(&[v, c] : g[u]) {
                q.push({v, du + c});
            }
        }
    }

    return -1;
}

int travel_plan(int _N, int _M, int _R[][2], int _L[], int _K, int _P[]) {
// globalization
    n = _N; ne = _M; nexit = _K;
    Rep (i, ne) g[_R[i][0]].push_back({_R[i][1], _L[i]}), g[_R[i][1]].push_back({_R[i][0], _L[i]});
    Rep (i, nexit) esc[i] = _P[i];
// end of globalization

    Rep (i, nexit) {
        q.push({esc[i], 0});
        q.push({esc[i], 0});
    }

    return Dijkstra();
}


#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...