#include "crocodile.h"
#include <bits/stdc++.h>
using namespace std;
int travel_plan(int n, int m, int edges[][2], int weights[], int exitCount, int exits[]){
#define ll long long
ll incoming[n];
multiset<ll> dist[n];
set<int> bruh;
memset(incoming, 0, sizeof(incoming));
for (ll x = 0; x < n; x++) dist[x].insert(LLONG_MAX/20),dist[x].insert(LLONG_MAX/20);
priority_queue<pair<ll, ll>, vector<pair<ll, ll>>, greater<pair<ll, ll>>> pq; //for those who have fulfilled the incoming >= 2 req
#define distUpdate(k, newval) dist[k].insert(newval); dist[k].erase(prev(dist[k].end()))
for (ll x = 0; x < exitCount; x++){
distUpdate(exits[x], 0);
distUpdate(exits[x], 0);
incoming[exits[x]] = INT_MIN;
pq.push({0, exits[x]});
}
vector<pair<ll, ll>> adjList[n];
for (ll x = 0; x < m; x++){
adjList[edges[x][0]].push_back({edges[x][1], weights[x]});
adjList[edges[x][1]].push_back({edges[x][0], weights[x]});
}
while (!pq.empty()){
ll d = pq.top().first, node = pq.top().second;
pq.pop();
//cout << node << ' ' << d << '\n';
if (d > *prev(dist[node].end()) || bruh.count(node) != 0) continue;
bruh.insert(node);
for (auto x : adjList[node]){
//cout << x.first << "upd\n";
ll prevSecond = *prev(dist[x.first].end());
if ( d + x.second < prevSecond){ //i.e. distUpdate was successful
incoming[x.first]++;
distUpdate(x.first, d + x.second);
if (incoming[x.first] >= 2) pq.push({*prev(dist[x.first].end()), x.first});
}
}
}
return *prev(dist[0].end());
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
596 KB |
Output is correct |
5 |
Correct |
1 ms |
596 KB |
Output is correct |
6 |
Correct |
1 ms |
468 KB |
Output is correct |
7 |
Correct |
1 ms |
596 KB |
Output is correct |
8 |
Correct |
1 ms |
596 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
596 KB |
Output is correct |
5 |
Correct |
1 ms |
596 KB |
Output is correct |
6 |
Correct |
1 ms |
468 KB |
Output is correct |
7 |
Correct |
1 ms |
596 KB |
Output is correct |
8 |
Correct |
1 ms |
596 KB |
Output is correct |
9 |
Correct |
2 ms |
724 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
468 KB |
Output is correct |
12 |
Correct |
3 ms |
852 KB |
Output is correct |
13 |
Correct |
3 ms |
980 KB |
Output is correct |
14 |
Correct |
1 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
468 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
340 KB |
Output is correct |
4 |
Correct |
1 ms |
596 KB |
Output is correct |
5 |
Correct |
1 ms |
596 KB |
Output is correct |
6 |
Correct |
1 ms |
468 KB |
Output is correct |
7 |
Correct |
1 ms |
596 KB |
Output is correct |
8 |
Correct |
1 ms |
596 KB |
Output is correct |
9 |
Correct |
2 ms |
724 KB |
Output is correct |
10 |
Correct |
1 ms |
340 KB |
Output is correct |
11 |
Correct |
1 ms |
468 KB |
Output is correct |
12 |
Correct |
3 ms |
852 KB |
Output is correct |
13 |
Correct |
3 ms |
980 KB |
Output is correct |
14 |
Correct |
1 ms |
340 KB |
Output is correct |
15 |
Correct |
1 ms |
468 KB |
Output is correct |
16 |
Correct |
361 ms |
70752 KB |
Output is correct |
17 |
Correct |
97 ms |
32332 KB |
Output is correct |
18 |
Correct |
145 ms |
33996 KB |
Output is correct |
19 |
Correct |
678 ms |
86164 KB |
Output is correct |
20 |
Correct |
208 ms |
52912 KB |
Output is correct |
21 |
Correct |
48 ms |
12096 KB |
Output is correct |
22 |
Correct |
249 ms |
53580 KB |
Output is correct |