# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
335939 | enerelt14 | Crocodile's Underground City (IOI11_crocodile) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "crocodile.h"
#include<bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define ff first
#define ss second
using namespace std;
vector<pair<int, int> >adj[100005];
int a[100005][3], s[100005];
int dijkstra(int k){
bool vis[100005];
priority_queue<pair<int, int> >pq;
pq.push(mp(0, k));
while(!pq.empty()){
int u = pq.top().ss;
int cost = -pq.top().ff;
if (s[u]>=2)return cost+a[u][2];
pq.pop();
path=min(cost, path);
vis[u] = 1;
for(int i = 0; i < adj[u].size(); i++){
int v = adj[u][i].ff;
int e = adj[u][i].ss;
if(!vis[v]) pq.push(mp(-(cost + e), v));
}
while(!pq.empty() && vis[pq.top().ss]) pq.pop();
}
}
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
for (int i=0;i<M;i++){
adj[R[i][0]].pb(mp(R[i][1], L[i]));
adj[R[i][1]].pb(mp(R[i][0], L[i]));
}
for (int i=0;i<K;i++){
for (int j=0;j<adj[P[i]].size();j++){
int v=adj[P[i]][j].ff;
int cost=adj[P[i]][j].ss;
s[v]++;
if (s[v]==1)a[v][1]=cost;
if (s[v]==2){
a[v][2]=max(a[v][1], cost);
a[v][1]=min(a[v][1], cost);
}
if (s[v]>=3){
if (cost<a[v][1]){
a[v][2]=a[v][1];
a[v][1]=cost;
}
if (cost>=a[v][1] && cost<a[v][2])a[v][2]=cost;
}
}
}
return dijkstra(0);
}