#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int MxN = 100010;
struct state_t {
int v;
ll w;
bool operator < (const state_t &o) const {
return w > o.w;
}
state_t(int _v, ll _w):
v(_v), w(_w) {}
};
vector<pair<int, ll>> adj[MxN];
priority_queue<state_t> pq;
ll dist[MxN][2]; // [0] = best, [1] = secondary
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]].emplace_back(R[i][1], L[i]);
adj[R[i][1]].emplace_back(R[i][0], L[i]);
}
memset(dist, 0x3f, sizeof dist);
for(int i=0; i<K; ++i) {
pq.emplace(P[i], dist[P[i]][0] = 0ll);
}
while(!pq.empty()) {
state_t now = pq.top(); pq.pop();
for(auto x: adj[now.v]) {
ll nxt = now.w + x.second;
if(dist[x.first][0] > nxt) {
if(dist[x.first][0] < 1e18 + 100) {
pq.emplace(x.first, dist[x.first][1] = dist[x.first][0]);
}
dist[x.first][0] = nxt;
}
else if(dist[x.first][1] > nxt){
pq.emplace(x.first, dist[x.first][1] = nxt);
}
}
}
return dist[0][1];
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
4180 KB |
Output is correct |
2 |
Correct |
2 ms |
4180 KB |
Output is correct |
3 |
Incorrect |
2 ms |
4180 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
4180 KB |
Output is correct |
2 |
Correct |
2 ms |
4180 KB |
Output is correct |
3 |
Incorrect |
2 ms |
4180 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
4180 KB |
Output is correct |
2 |
Correct |
2 ms |
4180 KB |
Output is correct |
3 |
Incorrect |
2 ms |
4180 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |