| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 751080 | ND0322 | Cyberland (APIO23_cyberland) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
typedef tuple<double,int,int> tu;
const double INF = DBL_MAX;
/*
void dfs(int node,vector<int> arr,int K,int H,vector<pii> adj[]){
visited[node] = true;
if(!arr[node]){
adj[0].push_back({node,0});
}
if(node == H){
return;
}
for(pii c:adj[node]){
if(!visited[c.first]){
dfs(c.first,arr,K,H,adj);
}
}
}
*/
double solve(int N, int M, int K, int H, std::vector<int> x, std::vector<int> y, std::vector<int> c, std::vector<int> arr){
vector<pii> adj[N];
K = min(K,70);
for(int i = 0; i < M; i++){
adj[x[i]].push_back({y[i],c[i]});
adj[y[i]].push_back({x[i],c[i]});
}
priority_queue<tu,vector<tu>,greater<tu>> pq;
vector<vector<double>>distances(N,vector<double>(K+1,INF));
//dfs(0,arr,K,H,adj);
pq.push({0.0,K,0});
distances[0][K] = 0.0;
while(!pq.empty()){
double d= get<0>(pq.top());
int node = get<2>(pq.top());
int k = get<1>(pq.top());
pq.pop();
if(d != distances[node][k] || node == H) continue;
/*
if(node == H){
continue;
}
*/
//d != distances[node][k] ||
for(pii c: adj[node]){
int child = c.first;
int weight = c.second;
if(!arr[child]){
if(distances[child][k]){
distances[child][k] = 0;
pq.push({0,k,child});
continue;
}
}
if(arr[child] == 2){
if(k && distances[child][k-1] > (d + 1.0 * weight)/2.0){
distances[child][k-1] = (d + weight)/2.0;
pq.push({distances[child][k-1],k-1,child});
}
}
if (distances[child][k] > d + 1.0 * weight){
distances[child][k] = d + 1.0 * weight;
pq.push({distances[child][k],k,child});
}
}
}
double ans = INF;
for(int i = 0; i <=K; i++){
ans = min(ans,distances[H][i]);
}
return (ans >= INF ? -1 : ans);
}
int main() {
//Im actually stupid
cout << solve(4, 4, 30, 3, {0, 0, 1, 2}, {1, 2, 3, 3}, {5, 4, 2, 4}, {1, 0, 2, 1}) << endl;
}