Submission #1265329

#	Time	Username	Problem	Language	Result	Execution time	Memory
1265329		canhnam357	Crocodile's Underground City (IOI11_crocodile)	C++20	89 / 100	325 ms	49076 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]) {
    vector<vector<pair<int, int>>> adj(N);
    for (int i = 0; i < M; i++) {
        adj[R[i][0]].emplace_back(R[i][1], L[i]);
        adj[R[i][1]].emplace_back(R[i][0], L[i]);
    }
    const int inf = 2e9 + 5;
    vector<vector<int>> dp(N, vector<int>(2, inf));
    priority_queue<pair<int, int>> q;
    for (int i = 0; i < K; i++) {
        dp[P[i]][0] = dp[P[i]][1] = 0;
        q.emplace(0, P[i]);
    }
    while (!q.empty()) {
        auto [d, u] = q.top();
        q.pop();
        d = -d;
        if (d != dp[u][1]) continue;
        for (auto [v, w] : adj[u]) {
            int fi = dp[v][0], se = dp[v][1];
            if (d + w < fi) {
                se = fi;
                fi = d + w;
            }
            else if (d + w < se) {
                se = d + w;
            }
            if (fi != dp[v][0] || se != dp[v][1]) {
                dp[v][0] = fi;
                dp[v][1] = se;
                q.emplace(-se, v);
            }
        }
    }
    return dp[0][1];
}

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