Submission #931027

#	Time	Username	Problem	Language	Result	Execution time	Memory
931027		tamir1	Cyberland (APIO23_cyberland)	C++17	8 / 100	3050 ms	7768 KiB

cyberland

#include "cyberland.h"
#include <bits/stdc++.h>
#include <vector>
#define ff first
#define ss second
using namespace std;
double ans;
vector<pair<int,int>> v[100010];
vector<int> ar;
bitset<100001> vis;
int h;
void dfs(int a,double dist,int k){
	if(vis[a]) return;
	vis[a]=1;
	if(a==h){
		ans=min(ans,dist);
		return;
	}
	for(pair<int,int> i:v[a]){
		if(ar[a]==0) dfs(i.ff,i.ss,k);
		if(ar[a]==1) dfs(i.ff,dist+i.ss,k);
		if(ar[a]==2){
			dfs(i.ff,dist+i.ss,k);
			if(k>0) dfs(i.ff,dist/2+i.ss,k-1);
		}
	}
	vis[a]=0;
}
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 i=0;i<M;i++){
		v[x[i]].push_back({y[i],c[i]});
		v[y[i]].push_back({x[i],c[i]});
	}
	h=H;
	ar=arr;
	ans=1e18;
	dfs(0,0,K);
	for(int i=0;i<N;i++){
		v[i].clear();
	}
	vis.reset();
    if(ans==1e18) return -1;
    return ans;
}

• 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