Submission #1239949

#	Time	Username	Problem	Language	Result	Execution time	Memory
1239949		countless	Crocodile's Underground City (IOI11_crocodile)	C++20	89 / 100	333 ms	77056 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

#define sp <<" "<<
#define endl "\n"

const ll INF = 1e18;

int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]) {
    vector<vector<pair<int, ll>>> adj(N);
    for (int i = 0; i < M; i++) {
        auto &[u, v] = R[i];
        adj[u].emplace_back(v, L[i]);
        adj[v].emplace_back(u, L[i]);
    }

    using state = tuple<ll, int>;
    vector<ll> dist1(N, INF), dist2(N, INF);
    priority_queue<state, vector<state>, greater<state>> pq;
    for (int i = 0; i < K; i++) {
        dist1[P[i]] = 0;
        dist2[P[i]] = 0;
        pq.emplace(0, P[i]);
    }

    while (!pq.empty()) {
        auto [d, u] = pq.top(); pq.pop();
        
        if (d > dist2[u]) continue;
        // cerr << d sp u << endl;
        
        for (auto &[v, w] : adj[u]) {
            ll n = d + w;
            if (n < dist1[v]) {
                dist2[v] = dist1[v];
                dist1[v] = n;
                pq.emplace(dist2[v], v);
            } else if (n < dist2[v]) {
                dist2[v] = n;
                pq.emplace(dist2[v], v);
            }

            // cerr << v sp dist1[v] sp dist2[v] << endl;
        }
    }

    return dist2[0];
}

• Subtask 1: 46 / 46
• Subtask 2: 43 / 43
• Subtask 3: 0 / 11