Submission #985046

#	Time	Username	Problem	Language	Result	Execution time	Memory
985046		rshohruh	Cyberland (APIO23_cyberland)	C++17	0 / 100	32 ms	9948 KiB

cyberland

#include "cyberland.h"

#include <vector>
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using node = pair<ll, ll>;


vector<vector<node>> g;
void dfs(int u, int p, vector<int>& res){
    if(u == p) return;
    res[u] = 1;
    for(auto [v, w]: g[u])
        if(!res[v])
            dfs(v, p, res);
}

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) {
    // for(int x: arr) assert(x < 2);
    g.assign(N, vector<node>());
    vector<ll> h(N, 1e18);
    for(int i = 0; i < M; ++i){
        g[x[i]].emplace_back(y[i], c[i]);
        g[y[i]].emplace_back(x[i], c[i]);
    }
    vector<int> up(N), to(N);
    dfs(0, H, up);
    dfs(H, 0, to);

    priority_queue<node, vector<node>, greater<node>> b; 
    for(int i = 0; i < N; ++i)
        if(up[i] & to[i])
            b.emplace(0, i);
    
    while(!b.empty()){
        auto [cur, u] = b.top();
        b.pop();
        if(h[u] <= cur) continue;
        h[u] = cur;
        for(auto [v, w]: g[u])
            b.emplace(cur+w, v);
    }
    
    return (h[H] == 1e18 ? -1 : h[H]);
}

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