Submission #1367276

#TimeUsernameProblemLanguageResultExecution timeMemory
1367276aaatroushCyberland (APIO23_cyberland)C++20
15 / 100
20 ms7236 KiB
#include "cyberland.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
#define ll long long
#define all(x) x.begin(), x.end()
#define pb push_back
#define pii pair<ll, ll>
#define in(v) for(auto &elem:v){ cin>>elem; }
#define out(v) for(auto elem:v){ cout<<elem<<" "; } cout<<endl;
#define FAST ios_base::sync_with_stdio(false); cin.tie(NULL);
typedef tree<pii, null_type, std::greater<>, 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> cc, vector<int> arr) {
    k = 30;
    vector<vector<pii>> adj(n);
    for(ll i = 0; i<m; i++){
        int u = x[i], v = y[i], c = cc[i];
        adj[u].pb({c, v});
        adj[v].pb({c, u});
    }

    vector<bool> vis(n, 0);
    queue<ll> q;
    q.push(0);
    vis[0] = 1;
    while(!q.empty()){
        ll u = q.front();
        q.pop();
        for (auto [t, v]:adj[u]){
            if(!vis[v]){
                q.push(v);
                vis[v] = 1;
            }
        }
    }
    const ll inf = 4e18;
    //out(vis)

    
    priority_queue<pii, vector<pii>, greater<pii>> pq;
    vector<ll> cost(n, inf);
    for(ll i = 0; i<n; i++){
        if (arr[i] == 0 && vis[i]){
            pq.push({0, i});
            cost[i] = 0;
        }
    }

    pq.push({0, 0});
    cost[0] = 0;
    while(!pq.empty()){
        auto [c, u] = pq.top();
        pq.pop();
        if(c > cost[u]){
            continue;
        }

        for(auto [c, v]:adj[u]){
            if(cost[v] <= cost[u] + c){
                continue;
            }
            
            cost[v] = cost[u] + c;
            pq.push({cost[v], v});
        }
    }
    if(cost[h] == inf){
        return -1;
    }
    //out(cost);
    return (long double)cost[h];
}
/*int main(){
    int n, m, h;
    cin>>n>>m>>h;
    vector<int> x(m), y(m), c(m), arr(n);
    for(ll i = 0; i<m; i++){
        cin>>x[i]>>y[i]>>c[i];
    }
    for(ll i = 0; i<n; i++){
        cin>>arr[i];
    }
    cout<<solve(n, m, 30, h, x, y, c, arr);
}*/
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...