제출 #983207

#제출 시각아이디문제언어결과실행 시간메모리
983207vjudge1사이버랜드 (APIO23_cyberland)C++17
0 / 100
28 ms6984 KiB
#include <vector>
#include <queue>
#include <climits>
using namespace std;

struct Edge {
    int to;
    int cost;
};

double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
    const int INF = INT_MAX;
    vector<vector<Edge>> graph(N);
    
    // Build the graph
    for (int i = 0; i < M; ++i) {
        graph[x[i]].push_back({y[i], c[i]});
        graph[y[i]].push_back({x[i], c[i]});
    }
    
    // Dijkstra's algorithm
    vector<int> dist(N, INF);
    priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
    dist[0] = 0;
    pq.push({0, 0});
    
    while (!pq.empty()) {
        int d = pq.top().first;
        int u = pq.top().second;
        pq.pop();
        
        if (dist[u] < d) continue;
        
        for (const auto& edge : graph[u]) {
            int v = edge.to;
            int w = edge.cost;
            
            // Apply special ability if available
            if (arr[v] == 2) {
                w /= 2;
                if (K > 0) {
                    w = min(w, dist[u]);
                    K--;
                }
            } else if (arr[v] == 0) {
                w = 0;
            }
            
            if (dist[v] > dist[u] + w) {
                dist[v] = dist[u] + w;
                pq.push({dist[v], v});
            }
        }
    }
    
    // Return the minimum time to reach Cyberland
    return dist[H] == INF ? -1 : dist[H];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...