# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
990668 | Ibrohim0704 | 사이버랜드 (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define int long long
double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
vector<double> min_time(N, numeric_limits<double>::infinity());
min_time[0] = 0;
// Custom comparator for the priority queue
auto cmp = [](const pair<double, int>& a, const pair<double, int>& b) {
return a.first > b.first;
};
priority_queue<pair<double, int>, vector<pair<double, int>>, decltype(cmp)> pq(cmp);
pq.push({0, 0});
while (!pq.empty()) {
double time = pq.top().first;
int country = pq.top().second;
pq.pop();
if (country == H) {
return time;
}
for (int i = 0; i < M; ++i) {
int neighbor;
if (x[i] == country) {
neighbor = y[i];
} else if (y[i] == country) {
neighbor = x[i];
} else {
continue;
}
double neighbor_time = time + c[i];
if (arr[neighbor] == 2 && K > 0) {
neighbor_time /= 2;
--K;
}
if (neighbor_time < min_time[neighbor]) {
min_time[neighbor] = neighbor_time;
pq.push({neighbor_time, neighbor});
}
}
}
return -1;
}