Submission #411459

#	Time	Username	Problem	Language	Result	Execution time	Memory
411459		AKaan37	Stations (IOI20_stations)	C++17	5 / 100	1200 ms	652 KiB

stations

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

using namespace std;

typedef long long lo; 
typedef pair< lo,lo > PII;

#define fi first
#define se second
#define mp make_pair
#define endl "\n"
#define pb push_back
#define fio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define FOR for(int i=1;i<=n;i++)
#define mid ((start+end)/2)
#define ort ((bas+son)/2)

const lo inf = 1000000000000000000;
const lo KOK = 100000;
const lo LOG = 30;
const lo li = 1005;
const lo mod = 1000000007;

vector<int> vec[li],vv;

inline void dfs(int node,int ata){
	//~ cout<<node<<endl;
	vv.pb(node);
	for(auto go:vec[node]){
		if(go==ata)continue;
		dfs(go,node);
	}
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> vvv) {
	vector<int> labels(n);
	vv.clear();
	for(int i=0;i<n;i++)vec[i].clear();
	for(int i=0;i<n-1;i++){
		vec[u[i]].pb(vvv[i]);
		vec[vvv[i]].pb(u[i]);
	}
	int ind=0;
	for(int i=0;i<n;i++){
		if((int)vec[i].size()==1)ind=i;
	}
	
	//~ cout<<ind<<endl;
	dfs(ind,-1);
	
	//~ for(auto go:vv)printf("%d **\n",go);
	for (int i = 0; i < n; i++) {
		//~ cout<<vv[i]<<endl;
		labels[vv[i]]=i;
		//~ labels.pb(vv[i]);
	}
	return labels;
}

int find_next_station(int s, int t, std::vector<int> c) {
	//~ while(1){s++;}
	//~ cout<<s<<" :: "<<t<<" :: "<<c[0]<<" :: "<<c[1]<<endl;
	if((int)c.size()==1)return c[0];
	if(s==t)return s;
	if(t>s){return c[1];}
	else return c[0];
}

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