Submission #434047

# Submission time Handle Problem Language Result Execution time Memory
434047 2021-06-20T14:26:39 Z kwongweng Stations (IOI20_stations) C++17
0 / 100
2501 ms 2097156 KB
#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
typedef vector<int> vi;
#define pb push_back

vi g[1000];
vi ans(1000), sz(1000), d(1000), used(1000);
int cur = 0;

void dfs1(int u = 0, int p = -1){
	for (int v : g[u]){
		if (v == p) continue;
		d[v] = d[u] + 1;
		dfs1(v, u);
		sz[u] += sz[v] + 1;
	}
}

void dfs2(int u = 0, int p = -1){
	if (d[u] % 2 == 0){
		used[cur] = 1;
		ans[u] = cur;
		while (used[cur]) cur++;
	}else{
		used[cur+sz[u]] = 1;
		ans[u] = cur+sz[u];
		while (used[cur]) cur++;
	}
	for (int v : g[u]){
		if (v == p) continue;
		dfs2(v, u);
	}
}

vi label(int n, int k, vi u, vi v) {
	ans.resize(n);
	for (int i = 0; i < n-1; i++){
		g[u[i]].pb(v[i]);
		g[v[i]].pb(u[i]);
	}
	dfs1();
	dfs2();
	return ans;
}

int find_next_station(int s, int t, vi c) {
	int C = c.size();
	if (C==1) return c[0];
	if (s < c[0]){
		if (t < s || t >= c[C-1]) return c[C-1];
		int pos = C-1;
		for (int i = C-2; i >= 0; i--){
			if (t < c[i+1] && t >= c[i]){
				pos = i; break;
			}
		}
		return c[pos];
	}
	if (t > s || t <= c[0]) return c[0];
	int pos = 0;
	for (int i = 1; i < C; i++){
		if (t > c[i-1] && t <= c[i]){
			pos = i; break;
		}
	}
	return c[pos];
}
# Verdict Execution time Memory Grader output
1 Runtime error 2196 ms 2097156 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 6 ms 636 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 1365 ms 2097156 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 994 ms 528 KB Output is correct
2 Runtime error 1132 ms 2097156 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 2501 ms 2097156 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -