Submission #939977

#	Time	Username	Problem	Language	Result	Execution time	Memory
939977		tamir1	Cyberland (APIO23_cyberland)	C++17	44 / 100	3061 ms	31312 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>
#include <vector>
#define ff first
#define ss second
#define ll long long
const double INF=1e18;
using namespace std;
ll h,i,j;
vector<pair<ll,double>> v[200001];
bitset<200005> vis;
priority_queue<pair<double,ll>> q;
double dp[200001][31],dist;
void dfs(ll a){
	if(a==h) return;
	if(vis[a]) return;
	vis[a]=1;
	for(pair<ll,double> l:v[a]){
		dfs(l.ff);
	}
}
void dijkstra(ll t){
	while(!q.empty()){
		ll a=q.top().ss;
		double d=q.top().ff;
		q.pop();
		if(dp[a][t]!=-1) continue;
		dp[a][t]=-d;
		if(a!=h) for(pair<ll,double> l:v[a]){
			if(dp[l.ff][t]==-1) q.push({d-l.ss,l.ff});
		}
	}
}
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) {
	h=H;
	dist=INF;
	for(i=0;i<M;i++){
		v[x[i]].push_back({y[i],c[i]});
		v[y[i]].push_back({x[i],c[i]});
	}
	dfs(0);
	q.push({0,0});
	for(i=0;i<N;i++){
		if(vis[i]==1 && arr[i]==0) q.push({0,i});
		for(j=0;j<=K;j++){
			dp[i][j]=-1;
		}
	}
	dijkstra(0);
	for(j=1;j<=K;j++){
		for(i=0;i<N;i++){
			if(vis[i]==1 && arr[i]==2) q.push({-dp[i][j-1],i});
		}
		dijkstra(j);
	}
	vis.reset();
	for(i=0;i<N;i++){
		v[i].clear();
	}
	for(j=0;j<=K;j++){
		if(dp[H][j]!=-1) dist=min(dist,dp[H][j]);
	}
	return (dist==INF ? -1 : dist);
}

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