Submission #795986

#TimeUsernameProblemLanguageResultExecution timeMemory
79598612345678Cyberland (APIO23_cyberland)C++17
97 / 100
3089 ms158364 KiB
#include "cyberland.h"

#include <bits/stdc++.h>

using namespace std;

const int nx=1e5+5, kx=71;

priority_queue<pair<double, pair<int, int>>, vector<pair<double, pair<int, int>>>, greater<pair<double, pair<int, int>>>> pq;
double dp[nx][kx], ans=DBL_MAX;

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<int, int>>> d(nx);
    K=min(K, 70);
    for (int i=0; i<M; i++) d[x[i]].push_back({y[i], c[i]}), d[y[i]].push_back({x[i], c[i]});
    for (int i=0; i<N; i++) for (int j=0; j<=K; j++) dp[i][j]=DBL_MAX;
    dp[0][0]=0;
    pq.push({0, {0, 0}});
    while (!pq.empty())
    {
        double cw=pq.top().first;
        int ck=pq.top().second.first, cl=pq.top().second.second;
        pq.pop();
        if (cl==H) continue;
        for (auto [v, w]:d[cl])
        {
            if (arr[v]==0&&dp[v][0]==DBL_MAX) pq.push({0, {0, v}}), dp[v][0]=0;
            if (dp[v][ck]>cw+w) pq.push({cw+w, {ck, v}}), dp[v][ck]=cw+w;
            if (arr[v]==2&&ck!=K) if (dp[v][ck+1]>(cw+w)/2) pq.push({(cw+w)/2, {ck+1, v}}), dp[v][ck+1]=(cw+w)/2; 
        }
    }
    ans=DBL_MAX;
    for (int i=0; i<=K; i++) ans=min(ans, dp[H][i]);
    if (ans==DBL_MAX) return -1;
    return ans;
}
#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...