Submission #832802

#	Time	Username	Problem	Language	Result	Execution time	Memory
832802		jlallas384	Crocodile's Underground City (IOI11_crocodile)	C++17	89 / 100	349 ms	62944 KiB

crocodile

#include <bits/stdc++.h>
#include "crocodile.h"
using namespace std;
using ll = long long;
struct edge{
    int u, v, w;
    int oth(int o){
        return u ^ v ^ o;
    }
};

int travel_plan(int n, int m, int r[][2], int L[], int k, int p[]){
    priority_queue<pair<int,int>> pq;
    vector<vector<int>> dist(n);
    for(int i = 0; i < k; i++){
        pq.emplace(0, p[i]);
        dist[p[i]].push_back(0);
        dist[p[i]].push_back(0);
    }
    vector<vector<pair<int,int>>> g(n);
    for(int i = 0; i < m; i++){
        g[r[i][0]].emplace_back(r[i][1], L[i]);
        g[r[i][1]].emplace_back(r[i][0], L[i]);
    }
    while(pq.size()){
        auto [d, v] = pq.top(); pq.pop();
        if(-d > dist[v].back()) continue;
        if(v == 0) return -d;
        for(auto [u, w]: g[v]){
            if(dist[u].empty()){
                dist[u].push_back(-d + w);
            }else if(dist[u].size() == 1){
                dist[u].push_back(-d + w);
                sort(dist[u].begin(), dist[u].end());
                pq.emplace(-dist[u][1], u);
            }else{
                if(-d + w < dist[u][1]){
                    dist[u].push_back(-d + w);
                    sort(dist[u].begin(), dist[u].end());
                    dist[u].pop_back();
                    pq.emplace(-dist[u][1], u);
                }
            }
        }
    }
}

Compilation message (stderr)

crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:13:35: warning: control reaches end of non-void function [-Wreturn-type]
   13 |     priority_queue<pair<int,int>> pq;
      |                                   ^~

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