제출 #1154255

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1154255		burgerguy	사이버랜드 (APIO23_cyberland)	C++20	44 / 100	3094 ms	30904 KiB

cyberland

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

vector<bool> isReachable;

void dfs(ll v, vector<vector<pair<ll,double>>>& adj, ll H) {
    if(v == H) return;

    for(pair<ll,ll> u : adj[v]) {
        if(!isReachable[u.first]) {
            isReachable[u.first] = true;
            dfs(u.first, adj, H);
        }
    }
}

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<vector<pair<ll,double>>> adj(N);
    for (int i = 0; i < M; i++) {
        adj[x[i]].emplace_back(y[i], c[i]);
        adj[y[i]].emplace_back(x[i], c[i]);
    }

    isReachable.clear();
    isReachable.resize(N);

    dfs(0, adj, H);
    isReachable[0] = true;

    if(!isReachable[H]) return -1;

    K = min(K, 60);

    priority_queue<tuple<double,ll,ll>, vector<tuple<double,ll,ll>>, greater<>> pq;
    vector<vector<double>> dist(N + 10, vector<double>(K + 10, 1e16));

    pq.emplace(0, 0, 0);
    dist[0][0] = 0;

    for(int i = 1; i < arr.size(); i++) {
        if(arr[i] == 0 && isReachable[i]) {
            pq.emplace(0, i, 0);
            dist[i][0] = 0;
        }
    }

    while(!pq.empty()) {
        double curDistance = get<0>(pq.top());
        ll curNode = get<1>(pq.top());
        ll curK = get<2>(pq.top());

        pq.pop();

        for(auto u : adj[curNode]) {
            double newDistance = curDistance + u.second;

            if(newDistance < dist[u.first][curK]) {
                dist[u.first][curK] = newDistance;
                pq.emplace(newDistance, u.first, curK);
            }

            if(arr[curNode] == 2 && curK < K) {
                newDistance /= 2;

                if(newDistance < dist[u.first][curK + 1]) {
                    dist[u.first][curK + 1] = newDistance;
                    pq.emplace(newDistance, u.first, curK + 1);
                }
            }
        }
    }

    double ans = 1e16;
    for (int i = 0; i <= K; ++i) {
        ans = min(ans, dist[H][i]);
    }

    return ans;
}

• Subtask 1: 0 / 5
• Subtask 2: 8 / 8
• Subtask 3: 13 / 13
• Subtask 4: 0 / 19
• Subtask 5: 7 / 7
• Subtask 6: 16 / 16
• Subtask 7: 0 / 29
• Subtask 8: 0 / 3