Submission #492971

#	Time	Username	Problem	Language	Result	Execution time	Memory
492971		1ne	Race (IOI11_race)	C++14	0 / 100	43 ms	70876 KiB

race

#include "race.h"
 
#include<bits/stdc++.h>
using namespace std;
vector<set<pair<int64_t,int64_t>>>adj(1e6);
int64_t k=0;
vector<int64_t>sub(1e6,0);
vector<int64_t>ans(1e6,0);
vector<int64_t>aux(1e6,0);
vector<bool>proceeded(1e6,0);
int64_t answer = INT_MAX;
int64_t kk = -1;
int64_t bookmark = 0;
void dfs (int64_t cur,int64_t par,int64_t cost,int64_t len,int64_t fill){
	if (cost>kk)return;
	if (fill){
		if (aux[cost]<bookmark){
			aux[cost] = bookmark;
			ans[cost] = len;
		}
		else if (len<ans[cost]){
			ans[cost] = len;
			aux[cost] = bookmark;
		}
	}
	else{
		if (aux[kk - cost]==bookmark){
			if (len + ans[kk - cost]<answer){
				answer = min(answer,len + ans[kk-cost]);
			}
			if (cost == kk){
				answer = min(answer,len);
			}
		}
	}
	for (auto x:adj[cur]){
		if (x.first!=par&&!proceeded[x.first]){
			dfs(x.first,cur,cost + x.second,len + 1,fill);
		}
	}
}
void getsub(int64_t u,int64_t par){
	sub[u]=1;
	k++;
	for (auto x:adj[u]){
		if (x.first!=par&&!proceeded[x.first]){
			getsub(x.first,u);
			sub[u]+=sub[x.first];
		}
	}
}
int64_t getcentroid(int64_t u,int64_t par){
	for (auto x:adj[u]){
		if (x.first!=par&&!proceeded[x.first]){
			if (sub[x.first]>k/2){
				return getcentroid(x.first,u);
			}
		}
	}
	return u;
}
void decomposition(int64_t u,int64_t par){
	k = 0;
	getsub(u,-1);
	//if (sub[u]<=1)return;
	int64_t c = getcentroid(u,-1);
	bookmark++;
	for (auto x:adj[c]){
		if (!proceeded[x.first]){
			dfs(x.first,c,x.second,1,0);
			dfs(x.first,c,x.second,1,1);
		}
	}
	proceeded[c]=true;
	for (auto x:adj[c]){
		if (!proceeded[x.first])
		//adj[x.first].erase({c,x.second});
		//adj[c].erase(x);
		decomposition(x.first,c);
	}
}
void build(){
	decomposition(0,-1);
}
int best_path(int N, int K, int H[][2], int L[])
{
	kk = K;
 	for (int64_t i = 0;i<N-1;++i){
 		adj[H[i][0]].insert({H[i][1],L[i]});
 		adj[H[i][1]].insert({H[i][0],L[i]});
 	}
 	build();
 	if (answer==INT_MAX){
 		return -1;
 	}
 	return answer;
}

• Subtask 1: 0 / 9
• Subtask 2: 0 / 12
• Subtask 3: 0 / 22
• Subtask 4: 0 / 57