Submission #725695

#	Time	Username	Problem	Language	Result	Execution time	Memory
725695		sofija6	Race (IOI11_race)	C++14	0 / 100	3 ms	4948 KiB

race

#include <bits/stdc++.h>
#include "race.h"
#define MAXN 200010
using namespace std;
vector<pair<int,int> > G[MAXN];
bool check[MAXN];
int subtree[MAXN],k,res=INT_MAX;
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;
}
vector<pair<int,int> > cur;
void DFS_Paths(int i,int p,int d,int cnt)
{
    if (d>k)
        return;
    cur.push_back({d,cnt});
    for (auto next : G[i])
    {
        if (next.first!=p)
            DFS_Paths(next.first,i,d+next.second,cnt+1);
    }
}
void Solve(int i)
{
    DFS_Subtree(i,i);
    int c=DFS_Centroid(i,i,i);
    check[c]=true;
    int ans[k+2];
    for (auto next : G[c])
    {
        cur.clear();
        DFS_Paths(next.first,c,next.second,1);
        for (auto i : cur)
        {
            if (!ans[k-i.first])
                continue;
            res=min(res,ans[k-i.first]+i.second);
        }
        for (auto i : cur)
            ans[i.first]=(!ans[i.first])?i.second : min(ans[i.first],i.second);
    }
    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);
}

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