Submission #403774

#	Time	Username	Problem	Language	Result	Execution time	Memory
403774		Newtech66	Race (IOI11_race)	C++17	100 / 100	1431 ms	43460 KiB

race

#include "race.h"
#include<bits/stdc++.h>
#define MAXN 200000
using namespace std;
vector<vector<pair<int,int> > > g(MAXN);
vector<int> subsz(MAXN);
bitset<MAXN> del;
int ans=INT_MAX;

void dfssz(int u,int p=-1)
{
    subsz[u]=1;
    for(auto ch:g[u])
    {
        if(ch.first==p) continue;
        if(!del.test(ch.first))
        {
            dfssz(ch.first,u);
            subsz[u]+=subsz[ch.first];
        }
    }
}

int getcen(int u,int sz,int p=-1)
{
    for(auto ch:g[u])
    {
        if(ch.first==p) continue;
        if(!del.test(ch.first))
            if(subsz[ch.first]>sz/2)    return getcen(ch.first,sz,u);
    }
    return u;
}

void place(map<int,int> &m,map<int,int> &m2,int len,int nm,int K)
{
    if(m2.find(len)==m2.end())    m2[len]=nm;
    else    m2[len]=min(m2[len],nm);
    if(m.find(K-len)!=m.end())    ans=min(ans,m2[len]+m[K-len]);
}

void solve(int u,map<int,int> &m,map<int,int> &m2,int len,int K,int dep,int p)
{
    for(auto ch:g[u])
    {
        if(ch.first==p) continue;
        if(!del.test(ch.first))
            if(len+ch.second<=K)
            {
                place(m,m2,len+ch.second,1+dep,K);
                solve(ch.first,m,m2,len+ch.second,K,1+dep,u);
            }
    }
}

void centroid(int u,const int& K)
{
    dfssz(u);
    int C=getcen(u,subsz[u]);
    map<int,int> m;
    for(auto ch:g[C])
    {
        if(!del.test(ch.first))
            if(ch.second<=K)
            {
                map<int,int> m2;
                place(m,m2,ch.second,1,K);
                solve(ch.first,m,m2,ch.second,K,1,C);
                for(auto p:m2)
                {
                    if(m.find(p.first)==m.end())    m[p.first]=p.second;
                    else    m[p.first]=min(m[p.first],p.second);
                }
            }
    }
    if(m.find(K)!=m.end())  ans=min(ans,m[K]);
    del.flip(C);
    for(auto ch:g[C])
        if(!del.test(ch.first)) centroid(ch.first,K);
}
 
int best_path(int N, int K, int H[][2], int L[])
{
    for(int i=0;i<N-1;i++)
    {
        int u=H[i][0],v=H[i][1];
        g[u].emplace_back(v,L[i]);
        g[v].emplace_back(u,L[i]);
    }
    centroid(0,K);
    if(ans==INT_MAX)  return -1;
    else  return ans;
}

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