제출 #1033512

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1033512		ArthuroWich	악어의 지하 도시 (IOI11_crocodile)	C++17	100 / 100	290 ms	71108 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[]) {
    int n = N, m = M;
    vector<int> vis(n, 0);
    vector<vector<pair<int, int>>> adj(n);
    for (int i = 0; i < m; i++) {
        adj[R[i][0]].push_back({R[i][1], L[i]});
        adj[R[i][1]].push_back({R[i][0], L[i]});
    }
    priority_queue<pair<int, int>> pq;
    for (int i = 0; i < K; i++) {
        vis[P[i]] = 1;
        pq.push({0, P[i]});
    }
    while(!pq.empty()) {
        auto [d, a] = pq.top();
        pq.pop();
        d = abs(d);
        vis[a]++;
        if (a == 0 && vis[a] > 1) {
            return d;
        }
        if (vis[a] == 1 || vis[a] > 2) {
            continue;
        }
        for (auto [b, w] : adj[a]) {
            if (vis[b] > 2) {
                continue;
            }
            pq.push({-(d+w), b});
        }
    }
    return 0;
}

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