Submission #1272976

#	Time	Username	Problem	Language	Result	Execution time	Memory
1272976		pvpro	Cyberland (APIO23_cyberland)	C++20	8 / 100	28 ms	6392 KiB

cyberland

#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>
#include <cstdio>
#include <vector>
#include <string>
#include <stack>
#include <queue>
#include <deque>
#include <bitset>
#include <algorithm>
#include <cmath>
#include <set>
#include <unordered_set>
#include <map>
#include <unordered_map>
#include <chrono>
#include <random>
#include <complex>
#include <numeric>
#include <assert.h>
#include <functional>
#include <climits>
#include <cstring>
#include <iterator>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

using ll = long long;
using ld = long double;
using ushort = unsigned short;

#ifndef LOCAL
#define endl "\n"
#endif
#define f first
#define s second
#define vec vector
#define pii pair<int, int>
#define all(x) x.begin(), x.end()
#define pb push_back
#define mp make_pair
template<typename T1, typename T2>
istream& operator>> (istream &in, pair<T1, T2> &p) { in >> p.f >> p.s; return in; }
template<typename T1, typename T2>
ostream& operator<< (ostream &out, pair<T1, T2> p) { out << "(" << p.f << " " << p.s << ")"; return out; }
#define printv(v) cerr << #v << " "; for (auto &x : v) { cerr << x << " "; } cerr << endl;
#define printvv(v) cerr << #v << " "; for (auto &x : v) { for (auto &y : x) { cerr << y << " "; } cerr << endl; }
#define debug(x) cerr << #x << " " << x << endl;

template<typename T>
istream& operator>> (istream &in, vector<T> &v) {
    for (auto &x : v) {
        in >> x;
    }
    return in;
}
template<typename T>
ostream& operator<< (ostream &out, vector<T> v) {
    for (auto &x : v) {
        out << x << " ";
    }
    return out;
}
template<typename T>
void operator-=(vector<T> &v, int delta) {
    for (auto &x : v) {
        x -= delta;
    }
}
template<typename T>
void operator+=(vector<T> &v, int delta) {
    for (auto &x : v) {
        x += delta;
    }
}

mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

double solve(int n, int m, int k, int h, vector<int> x, vector<int> y, vector<int> c, vector<int> arr) {
    vector<vector<pii>> graph(n);
    for (int i = 0; i < m; ++i) {
        graph[x[i]].pb(mp(y[i], c[i]));
        graph[y[i]].pb(mp(x[i], c[i]));
    }
    priority_queue<pii> q;
    vector<double> d(n, 1e18);
    vector<int> marked(n);
    q.emplace(0, 0);
    d[0] = 0;
    for (int i = 1; i < n; ++i) {
        if (arr[i] == 0) {
            q.emplace(0, i);
            d[i] = 0;
        }
    }
    while (!q.empty()) {
        int v = q.top().s;
        q.pop();
        if (marked[v]) {
            continue;
        }
        marked[v] = true;
        for (auto &pu : graph[v]) {
            if (d[pu.f] > d[v] + pu.s) {
                d[pu.f] = d[v] + pu.s;
                q.emplace(-d[pu.f], pu.f);
            }
        }
    }
    return d[h];
}

#ifdef LOCAL
int main() {
    freopen("in.txt", "r", stdin);
    int T;
    assert(1 == scanf("%d", &T));
    while (T--){
        int N,M,K,H;
        assert(4 == scanf("%d %d %d\n%d", &N, &M, &K, &H));
        std::vector<int> x(M);
        std::vector<int> y(M);
        std::vector<int> c(M);
        std::vector<int> arr(N);
        for (int i=0;i<N;i++)
            assert(1 == scanf("%d", &arr[i]));
        for (int i=0;i<M;i++)
            assert(3 == scanf("%d %d %d", &x[i], &y[i], &c[i]));
        printf("%.12lf\n", solve(N, M, K, H, x, y, c, arr));
    }
}
#endif

• Subtask 1: 0 / 5
• Subtask 2: 8 / 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