Submission #424089

#	Time	Username	Problem	Language	Result	Execution time	Memory
424089		lakshith_	Stations (IOI20_stations)	C++14	21 / 100	1119 ms	740 KiB

stations

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

using namespace std;

#define what_is(a) cout << #a << " is " << a << "\n"
#define checker(a) cout <"checker reached " << a << "\n"
#define ll long long

vector<vector<int>> adj(1000,vector<int>());

vector<int> lbl;

void dfs(int u,int p,int d){
	lbl[u]=d++;
	for(int v:adj[u])
		if(v!=p)
			dfs(v,u,d);
}


std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
	lbl = vector<int>(n);
	for(int i=0;i<n;i++)
		adj[i] = vector<int>();
	for(int i=0;i<n-1;i++){
		adj[u[i]].push_back(v[i]);
		adj[v[i]].push_back(u[i]);
	}
	int src = -1;
	for(int i=0;i<n;i++)
		if(adj[i].size()>2)src = i;
	if(src==-1)
		for(int i=0;i<n;i++)
			if(adj[i].size()==1)src = i;
	lbl[src]=0;
	int d = 1;
	for(int a:adj[src]){
			dfs(a,src,d);
			d+=1000;
	}
	return lbl;
}

int find_next_station(int s, int t, std::vector<int> c) {
	if(t==0)return c[0];
	if(s==0)return ((ll)t/1000LL)*1000LL + 1LL;
	if(s/1000==t/1000){
		if(s<t)return c[1];
		else return c[0];
	}else{
		return c[0];
	}
}

• Subtask 1: 5 / 5
• Subtask 2: 0 / 8
• Subtask 3: 16 / 16
• Subtask 4: 0 / 10
• Subtask 5: 0 / 61