#include <bits/stdc++.h>
using namespace std;
//#define int long long
const long long INF = 1e18;
int travel_plan(int n, int m, int r[][2], int l[], int k, int p[]){
vector< vector< pair<long long, long long> > > Graph(n);
for(int i = 0; i < m; i++){
Graph[r[i][0]].push_back(make_pair(l[i], r[i][1]));
Graph[r[i][1]].push_back(make_pair(l[i], r[i][0]));
}
priority_queue< pair<long long, long long>, vector< pair<long long, long long> >, greater< pair<long long, long long> > > pq;
vector< vector< long long > > dist(n, vector<long long>(2, INF));
for(int i = 0; i < k; i++){
dist[p[i]][0] = dist[p[i]][1] = 0;
pq.push(make_pair(0,p[i]));
}
while(!pq.empty()){
pair<long long, long long> pa = pq.top();
pq.pop();
long long cost = pa.first;
int node = pa.second;
for(auto a : Graph[node]){
if(dist[a.second][0] > a.first + cost){
dist[a.second][1] = dist[a.second][0];
dist[a.second][0] = a.first+cost;
pq.push(make_pair(dist[a.second][1], a.second));
}else if(dist[a.second][1] > a.first+cost){
dist[a.second][1] = a.first+cost;
pq.push(make_pair(dist[a.second][1], a.second));
}
}
}
return dist[0][1];
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Incorrect |
3 ms |
384 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Incorrect |
3 ms |
384 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
384 KB |
Output is correct |
2 |
Correct |
3 ms |
384 KB |
Output is correct |
3 |
Incorrect |
3 ms |
384 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |