# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
749116 | gun_gan | 사이버랜드 (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
// #include <bits/stdc++.h>
#include "cyberland.h"
using namespace std;
typedef long long ll;
double solve(int N, int M, int K, int H, vector<int> x, vector<int>
y, vector<int> c, vector<int> arr) {
vector<vector<pair<int,int>>> g(N);
for(int i = 0; i < M; i++) {
g[x[i]].push_back({y[i], c[i]});
g[y[i]].push_back({x[i], c[i]});
}
priority_queue<pair<ll,ll>> pq;
vector<ll> d(N, 1e18);
d[0] = 0;
pq.push({0, 0});
for(int i = 1; i < N; i++) {
if(!arr[i]) {
d[i] = 0;
pq.push({0, i});
}
}
while(!pq.empty()) {
auto [x, v] = pq.top(); pq.pop();
if(d[v] < x) continue;
for(auto [u, w] : g[v]) {
if(d[u] > x + w) {
d[u] = x + w;
pq.push({d[u], u});
}
}
}
double res = d[H];
if(res == 1e18) res = -1;
return res;
}
// int main() {
// cin.tie(0); ios_base::sync_with_stdio(0);
// cout << fixed << setprecision(1) << solve(3, 1, 30, 2, {1}, {2}, {12}, {1, 2, 1}) << '\n';
// }