Submission #1080344

#	Time	Username	Problem	Language	Result	Execution time	Memory
1080344		KasymK	Cyberland (APIO23_cyberland)	C++17	68 / 100	2969 ms	39012 KiB

cyberland

#include "bits/stdc++.h"
using namespace std;
#define ff first
#define ss second   
#define all(v) v.begin(), v.end()
#define ll long long
#define pb push_back
#define pii pair<int, int>
#define pli pair<ll, int>
#define pll pair<ll, ll>
#define tr(i, c) for(auto i = c.begin(); i != c.end(); ++i)
#define wr puts("----------------")
template<class T>bool umin(T& a,T b){if(a>b){a=b;return 1;}return 0;}
template<class T>bool umax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
const int N = 2e5+5;
const int K = 70;
vector<pair<int, double>> adj[N];
bool vis0[N], vis[N][K];
priority_queue<tuple<int, double, int>, vector<tuple<int, double, int>>, greater<tuple<int, double, int>>> q;
vector<int> ad;

void dfs(int x){
    if(vis0[x])
        return; 
    vis0[x] = 1;
    if(ad[x] ==  0)
        q.push({0, 0, x});
    for(auto [i, w] : adj[x])
        dfs(i);
}

double solve(int n, int m, int k, int h, vector<int> u, vector<int> v, vector<int> w, vector<int> arr){
    umin(k, 70);
    ad.clear();
    for(auto &i : arr)
        ad.pb(i);
    memset(vis0, 0, sizeof vis0);
    for(int i = 0; i <= n; ++i)
        adj[i].clear();
    while(!q.empty())
        q.pop();
    ad[0] = 0;
    vis0[h] = 1;
    for(int i = 0; i < m; ++i){
        adj[u[i]].pb({v[i], w[i]});
        adj[v[i]].pb({u[i], w[i]});
    }
    double answer = 1e18;
    dfs(0);
    memset(vis, 0, sizeof vis);
    while(!q.empty()){
        auto [x0, w0, x] = q.top();
        q.pop();
        if(x == h){
            umin(answer, w0);
            continue;
        }
        if(vis[x][x0])
            continue;
        vis[x][x0] = 1;
        for(auto &[i, w] : adj[x]){
            if(!vis[i][x0])
                q.push({x0, w0+w, i});
            if(ad[i] == 2 and x0 < K and !vis[i][x0+1])
                q.push({x0+1, (w0+w)/(double)(2), i});
        }
    }
    if(answer == 1e18)
        return -1;
    else
        return answer;
}

// int main(){
//     double kk = solve(3, 2, 30, 2, {1, 2}, {2, 0}, {12, 4}, {1, 2, 1});
//     double kk1 = solve(4, 4, 30, 3, {0, 0, 1, 2}, {1, 2, 3, 3}, {5, 4, 2, 4}, {1, 0, 2, 1});
//     if(kk == 4.0 and kk1 == 2.0)
//         puts("Correct.");
//     else
//         puts("Wrong Answer.");
//     return 0;
// }

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