Submission #759576

#	Time	Username	Problem	Language	Result	Execution time	Memory
759576		IvanJ	Cyberland (APIO23_cyberland)	C++17	15 / 100	3051 ms	12116 KiB

cyberland

#include "cyberland.h"
#include<bits/stdc++.h>

#define pb push_back
#define x first
#define y second
#define all(a) (a).begin(), (a).end()

using namespace std;

typedef long long ll;
typedef pair<int, int> ii;

const int maxn = 1e5 + 5;
const double inf = 1e18;
double dist[maxn];
vector<ii> adj[maxn];
int vis[maxn];

void reach(int x) {
	if(vis[x]) return;
	vis[x] = 1;
	for(auto p : adj[x]) 
		reach(p.x);
}

double solve(int N, int M, int K, int H, vector<int> u, vector<int> v, vector<int> c, vector<int> arr) {
	for(int i = 0;i < N;i++) {
		adj[i].clear();
		vis[i] = 0;
		dist[i] = inf;
	}
	
	for(int i = 0;i < M;i++) 
		adj[u[i]].pb({v[i], c[i]}), adj[v[i]].pb({u[i], c[i]});
	
	reach(0);
	for(int i = 0;i < N;i++) 
		if(arr[i] == 0 && vis[i] == 1) dist[i] = 0;
	dist[0] = 0;
	
	for(int k = 0;k <= K;k++) {
		set<pair<double, int>> s;
		for(int i = 0;i < N;i++) 
			if(vis[i] == 1) s.insert({dist[i], i});
		
		vector<int> change(N, 0);
		while(s.size()) {
			auto state = *s.begin();
			s.erase(state);
			int x = state.y;
			if(x == H) break;
			
			for(auto p : adj[x]) {
				int y = p.x;
				double cost = p.y;
				
				if(dist[x] + cost < dist[y]) {
					s.erase({dist[y], y});
					dist[y] = dist[x] + cost;
					change[y] = 1;
					s.insert({dist[y], y});
				}
			}
		}
		
		for(int i = 0;i < N;i++)
			if(arr[i] == 2 && vis[i] == 1 && change[i] == 1) dist[i] /= 2.0;
	}
	
	if(dist[H] == inf) return -1;
	return dist[H]; 
}

• 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