Submission #756826

#	Time	Username	Problem	Language	Result	Execution time	Memory
756826		Valters07	Cyberland (APIO23_cyberland)	C++17	15 / 100	3037 ms	39232 KiB

cyberland

#include <bits/stdc++.h>
#include "cyberland.h"
#define ll long long
#define pb push_back
#define fi first
#define se second
using namespace std;
const int N = 1e5+5;
const int K = 30+2;
vector<pair<int,int> > g[N];
ll dist[N][K];
bool vis[N][K];
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<m;i++)
        g[x[i]].pb({y[i],c[i]}),
        g[y[i]].pb({x[i],c[i]});
    for(int i = 0;i<n;i++)
        for(int j = 0;j<=k;j++)
            dist[i][j]=1e18,
            vis[i][j]=0;
    priority_queue<pair<long double,pair<int,int> >,vector<pair<long double,pair<int,int> > >,greater<pair<long double,pair<int,int> > > > q;
    dist[0][0]=0;
    q.push({0,{0,0}});
    for(int i = 0;i<n;i++)
        if(arr[i]==0)
            dist[i][0]=0,
            q.push({0,{i,0}});
    for(int i = 0;i<=k;i++)
        vis[h][i]=1;
    while(!q.empty())
    {
        int u = q.top().se.fi, dv = q.top().se.se;
        q.pop();
        if(vis[u][dv])
            continue;
        vis[u][dv]=1;
        for(auto v:g[u])
        {
            long double nwc = dist[u][dv]+v.se;
            if(dist[v.fi][dv]>nwc)
                dist[v.fi][dv]=nwc,
                q.push({nwc,{v.fi,dv}});
            if(arr[v.fi]==2&&dv<k)
            {
                nwc/=2;
                if(dist[v.fi][dv+1]>nwc)
                    dist[v.fi][dv+1]=nwc,
                    q.push({nwc,{v.fi,dv+1}});
            }
        }
    }
    ll best = 1e18;
    for(int i = 0;i<=k;i++)
        best=min(best,dist[h][i]);
    for(int i = 0;i<n;i++)
        g[i].clear();
    if(best==1e18)
        return -1;
    return best;
}

• 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