Submission #543215

#	Time	Username	Problem	Language	Result	Execution time	Memory
543215		__Variatto	Crocodile's Underground City (IOI11_crocodile)	C++17	100 / 100	525 ms	67444 KiB

crocodile

#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define ll long long
const int MAX=1e6+10, MAXV=1e9+100;
int n, m, k, ileSpe[MAX], odw[MAX];
int odl[MAX];
bool spe[MAX];
vector<pair<int,int>>g[MAX];
priority_queue<pair<int,int>>kol;
pair<int,int>akt[MAX];
int obl(int x){
    return(akt[x].se);
}
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[]){
    n=N, k=K, m=M;
    for(int i=1; i<=m; i++)
        g[R[i-1][0]+1].pb({R[i-1][1]+1, L[i-1]}), g[R[i-1][1]+1].pb({R[i-1][0]+1, L[i-1]});
    for(int i=1; i<=k; i++){
        int v=P[i-1]+1;
        spe[v]=true;
    }
    for(int i=1; i<=n; i++){
        odl[i]=MAXV;
        akt[i]={MAXV, MAXV};
        if(spe[i])
            kol.push({0, i}), odl[i]=0;
    }
    while(kol.size()){
        auto v=kol.top().se;
        kol.pop();
        if(odw[v]) continue;
        odw[v]=spe[v]=true;
        for(auto s: g[v]){
            ileSpe[s.fi]++;
            if(odl[v]+s.se<=akt[s.fi].fi)
                akt[s.fi].se=akt[s.fi].fi, akt[s.fi].fi=odl[v]+s.se;
            else if(odl[v]+s.se<akt[s.fi].se)
                akt[s.fi].se=odl[v]+s.se;
            if(ileSpe[s.fi]>=2){
                int nowy=obl(s.fi);
                if(odl[s.fi]>nowy)
                    odl[s.fi]=nowy, kol.push({-nowy, s.fi});
            }
        }
    }
    return odl[1];
}

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