# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
983351 | crispxx | 사이버랜드 (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef long long ll;
typedef long double ld;
typedef tree <int, null_type, less<int>, rb_tree_tag,
tree_order_statistics_node_update> ordered_set;
typedef pair<double, int> pii;
#define nl '\n'
#define vt vector
#define ar array
#define pb push_back
#define ff first
#define ss second
const int INF = INT_MAX;
double solve(int N, int M, int K, int H, vt<int> x, vt<int> y, vt<int> c, vt<int> arr) {
int n = N, m = M, k = K, h = H;
vt<pair<int, int>> adj[n];
for(int i = 0; i < m; i++) {
adj[x[i]].pb({y[i], c[i]});
adj[y[i]].pb({x[i], c[i]});
}
vector<double> d(n, numeric_limits<double>::max());
priority_queue<pii, vt<pii>, greater<pii>> q;
d[0] = 0;
q.push({0, 0});
while(!q.empty()) {
int u = q.top().ss;
double res = q.top().ff;
q.pop();
if(u == h) return res;
int ch = k;
if(arr[u] == 2) ch--;
for(auto &edge : adj[u]) {
int v = edge.ff;
int c = edge.ss;
double nw = res + c;
if(arr[v] == 0) nw = res;
if(arr[v] == 2 && ch--)
nw = res + c / 2.;
if(nw =< d[v]) {
d[v] = nw;
q.push({nw, v});
}
}
}
return -1;
}