Submission #750428

#	Time	Username	Problem	Language	Result	Execution time	Memory
750428		Gurban	Cyberland (APIO23_cyberland)	C++17	15 / 100	3053 ms	8184 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>

using namespace std;

using ld = long double;

const double inf = 1e16;
const int maxn=1e5+5;
const double eps = 0.00000001;
int D[maxn];
vector<pair<int,double>>E[maxn];

double solve(int N, int M, int K, int H, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
    for(int i = 0;i < N;i++) E[i].clear();
    for(int i = 0;i < M;i++){
        E[x[i]].push_back({y[i],c[i]});
        E[y[i]].push_back({x[i],c[i]});
    }
    for(int i = 1;i < N;i++) D[i] = 1e9;
    D[0] = 0;
    deque<int>deq;
    vector<bool>vis(N);
    deq.push_back(0);
    while(!deq.empty()){
        int x = deq.front();
        deq.pop_front();
        if(vis[x]) continue;
        vis[x] = 1;
        for(auto i : E[x]){
            if(D[x] + (arr[i.first] == 2) < D[i.first]){
                D[i.first] = D[x] + (arr[i.first] == 2);
                if(arr[i.first] != 2) deq.push_front(i.first);
                else deq.push_back(i.first);
            }
        }
    }
    vector<int>zeros = {0};
    vector<int>two;
    for(int i = 1;i < N;i++) if(!arr[i]) zeros.push_back(i);
    for(int i = 1;i < N;i++) if(arr[i] == 2) two.push_back(i);
    priority_queue<pair<double,int>>q;
    q.push({0,H});
    double ans = inf;
    double der = 1;
    for(int i = 0;i <= K;i++){
        vector<double>dis(N,inf);
        vector<bool>vis(N,0);
        while(!q.empty()){
            int x = q.top().second;
            double len = -q.top().first;
            q.pop();
            if(vis[x]) continue;
            vis[x] = 1;
            for(auto j : E[x]){
                double now = (j.second / der) + len;
                if(dis[j.first] > now){
                    dis[j.first] = now;
                    if(arr[j.first] != 2 and j.first != H) q.push({-now,j.first}); 
                }
            }
        }
        for(auto j : zeros) if(D[j] + i <= K) ans = min(ans,dis[j]);
        for(auto j : two) if(abs(dis[j] - inf) >= eps) q.push({-dis[j],j});
        der *= 2;
    }
    if(abs(ans - inf) < eps) return -1;
    return ans;
}

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