Submission #861339

#	Time	Username	Problem	Language	Result	Execution time	Memory
861339		8pete8	Race (IOI11_race)	C++14	0 / 100	6 ms	11612 KiB

race

#include<iostream>
#include<stack>
#include<map>
#include<vector>
#include<string>
#include<unordered_map>
#include <queue>
#include<cstring>
#include<limits.h>
#include<cmath>
#include<set>
#include<algorithm>
#include<bitset> 
#include<stack>
using namespace std;
#define ll long long
#define f first
#define endl "\n"
#define s second
#define pii pair<ll,ll>
#define pppiiii pair<pii,pii>
#define ppii pair<ll,pii>
#define all(x) x.begin(),x.end()
#define pb push_back
#define mp make_pair
#define lb lower_bound
#define ub upper_bound
#define fastio ios::sync_with_stdio(false);cin.tie(NULL);
const int mxn=1e5+1,mx=1e6;
ll inf=1e18;
vector<pii>adj[mxn+10];
ll sz[mxn+10],n,k;
ll mn[mx+10],dist[mxn+10];
ll ans=inf;
bitset<mxn+10>del;
void dfs(int cur,int p){
    sz[cur]=1;
    for(auto i:adj[cur])if(i.f!=p&&(!del[i.f]))dfs(i.f,cur),sz[cur]+=sz[i.f];
}
int getcen(int cur,int p){
	for(auto i:adj[cur]){
		if(i.f==p||del[i.f])continue;
		if(sz[i.f]*2>=n)return getcen(i.f,cur);
	}
	return cur;
}
void solve(int cur,int p,ll d,ll di){
    if(di>k)return;
    dist[di]=min(dist[di],d);
    if(mn[k-di]!=inf)ans=min(ans,mn[k-di]+dist[di]);
    for(auto i:adj[cur]){
        if(i.f==p||del[i.f])continue;
        solve(i.f,cur,d+1,di+i.s);
    }
}
void decomp(int cur){
    fill(mn,mn+k+1,inf);
    fill(dist,dist+k+1,inf);
    fill(sz,sz+mxn+1,1);
    dfs(cur,-1);
    int node=getcen(cur,-1);
    dist[0]=0;
    mn[0]=0;
    for(auto i:adj[node]){
        if(del[i.f])continue;
        solve(i.f,node,1,i.s);
        for(int i=0;i<=k;i++)mn[i]=min(mn[i],dist[i]);
    }
    del[node]=true;
    for(auto i:adj[node])if(!del[i.f])decomp(i.f);
}
int best_path(int N, int K, int H[][2], int L[]){
	n=N;
    k=K;
	for(int i=0;i<N-1;i++){
		adj[H[i][0]].pb({H[i][1],L[i]});
		adj[H[i][1]].pb({H[i][0],L[i]});
	}
    decomp(0);
    if(ans==inf)ans=-1;
    return ans;
}/*
int main(){
    cin>>n>>k;
    int h[n][2],l[n];
    for(int i=0;i<n-1;i++)cin>>h[i][0]>>h[i][1]>>l[i];
    cout<<best_path(n,k,h,l);
}*/

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