Submission #412185

#	Time	Username	Problem	Language	Result	Execution time	Memory
412185		AKaan37	Stations (IOI20_stations)	C++17	21 / 100	1097 ms	740 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;

int sayac=0,mx;
vector<int> vec[li],vv;
int dp[li];

inline void dfs1(int node,int ata,int der){
	mx=max(mx,der);
	for(auto go:vec[node]){
		if(go==ata)continue;
		dfs1(go,node,der+1);
	}
}

inline void dfs(int node,int ata){
	vv[node]=sayac++;
	int flagg=0;
	//~ vv[node]+=(1<<node);
	for(auto go:vec[node]){
		if(go==ata)continue;
		flagg=1;
		dfs(go,node);
		//~ vv[node]+=vv[go];
	}
	if(flagg==0)sayac+=1000;
}

std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> vvv) {
	std::vector<int> labels(n);
	sayac=0;
	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]);
	}
	vv.clear();
	int ind=-1;
	for(int i=0;i<n;i++){
		vv.pb(0);
		if((int)vec[i].size()>2)ind=i;
	}
	if(ind==-1){
		for(int i=0;i<n;i++){
			if((int)vec[i].size()==1)ind=i;
		}
	}
	//~ cout<<ind
	dfs1(ind,-1,0);
	//~ mx=max(mx,1000-1);
	dfs(ind,-1);
	for (int i = 0; i < n; i++) {
		//~ cout<<vv[i]<<" ** "<<endl;
		labels[i] = vv[i];
	}
	return labels;
}

inline int f(int sira,int hedef){
	int cevv=0;
	if(sira>hedef)return 0;
	if(sira==hedef)return 1;
	if(~dp[sira])return dp[sira];
	cevv=max(cevv,f(sira*2+1,hedef));
	cevv=max(cevv,f(sira*2+2,hedef));
	return dp[sira]=cevv;
}

int find_next_station(int s, int t, std::vector<int> c) {
	//~ for(int i=0;i<(int)c.size()-1;i++){
		//~ if((c[i]&t)==t)return c[i];
	//~ }
	//~ return c[(int)c.size()-1];
	if(t==0)return c[0];
	if((int)c.size()==1)return c[0];
	if(t<s)return c[0];
	if(abs(t-s)>=1000 && (int)c.size()==2)return c[0];
	if((int)c.size()==2)return c[1];
	int siz=(int)c.size();
	for(int i=0;i<siz-1;i++){
		if(c[i+1]>t)return c[i];
	}
	return c[siz-1];
}

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