Submission #725920

#TimeUsernameProblemLanguageResultExecution timeMemory
725920sofija6Race (IOI11_race)C++14
100 / 100
481 ms37512 KiB
#include <bits/stdc++.h>
#include "race.h"
#define MAXN 200010
#define MAXK 1000010
using namespace std;
vector<pair<int,int> > G[MAXN];
bool check[MAXN];
int subtree[MAXN],k,res=INT_MAX;
pair<int,int> ans[MAXK];
void DFS_Subtree(int i,int p)
{
    subtree[i]=1;
    for (auto next : G[i])
    {
        if (next.first!=p && !check[next.first])
        {
            DFS_Subtree(next.first,i);
            subtree[i]+=subtree[next.first];
        }
    }
}
int DFS_Centroid(int i,int p,int root)
{
    for (auto next : G[i])
    {
        if (next.first!=p && !check[next.first] && subtree[next.first]>subtree[root]/2)
            return DFS_Centroid(next.first,i,root);
    }
    return i;
}
int c;
void DFS_Paths(int i,int p,int d,int cnt)
{
    if (d>k)
        return;
    if ((ans[k-d].first && ans[k-d].second==c) || k==d)
        res=min(res,ans[k-d].first+cnt);
    for (auto next : G[i])
    {
        if (next.first!=p && !check[next.first])
            DFS_Paths(next.first,i,d+next.second,cnt+1);
    }
}
void DFS_Fill(int i,int p,int d,int cnt)
{
    if (d>k)
        return;
    if (d && (!ans[d].first || ans[d].second!=c || ans[d].first>cnt))
        ans[d]={cnt,c};
    for (auto next : G[i])
    {
        if (next.first!=p && !check[next.first])
            DFS_Fill(next.first,i,d+next.second,cnt+1);
    }
}
void Solve(int i)
{
    DFS_Subtree(i,i);
    c=DFS_Centroid(i,i,i);
    check[c]=true;
    for (auto next : G[c])
    {
        if (check[next.first])
            continue;
        DFS_Paths(next.first,c,next.second,1);
        DFS_Fill(next.first,c,next.second,1);
    }
    for (auto next : G[c])
    {
        if (!check[next.first])
            Solve(next.first);
    }
}
int best_path(int N, int K, int H[][2], int L[])
{
    k=K;
    for (int i=0;i<N-1;i++)
    {
        G[H[i][0]].push_back({H[i][1],L[i]});
        G[H[i][1]].push_back({H[i][0],L[i]});
    }
    Solve(0);
    return (res==INT_MAX?-1 : res);
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...