#include <bits/stdc++.h>
#include "crocodile.h"
using namespace std;
int travel_plan(int N, int M, int R[][2], int L[], int K, int P[])
{
int n=N, m=M, k=K; vector<pair<long long, int>>adj[n]; vector<int> dis(n, -1);
vector<bool>exit(n, false); vector<long long>d(n, 1e9); queue<int>q;
priority_queue<pair<long long, int>, vector<pair<long long, int>>, greater<pair<long long, int>>>pq;
for(int l=0;l<m;l++) {
int a=R[l][0], b=R[l][1]; long long w=L[l];
adj[a].push_back({b, w});
adj[b].push_back({a, w});
}
for(int l=0;l<k;l++) {
int a=P[l]; dis[a]=0;
pq.push({0, a}); d[a]=0; q.push(a);
}
vector<bool>queued(n, false);
vector<long long>len[n]; vector<int>income(n, 0);
while(!pq.empty()) {
int a=pq.top().second; long long w=pq.top().second;
pq.pop(); queued[a]=true;
for(auto u:adj[a]) {
int to=u.first; long long s=u.second;
if(queued[to]) continue;
len[to].push_back(d[a]+s); income[to]++;
if(income[to]>1) {
sort(len[to].begin(), len[to].end());
pq.push({len[to][1], to});
d[to]=len[to][1];
}
}
}
return d[0];
}
Compilation message
crocodile.cpp: In function 'int travel_plan(int, int, int (*)[2], int*, int, int*)':
crocodile.cpp:23:42: warning: unused variable 'w' [-Wunused-variable]
23 | int a=pq.top().second; long long w=pq.top().second;
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
204 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
204 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
204 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |