Submission #585255

#	Time	Username	Problem	Language	Result	Execution time	Memory
585255		MrDeboo	Race (IOI11_race)	C++17	43 / 100	332 ms	103568 KiB

race

#include "race.h"
#include <bits/stdc++.h>
using namespace std;
vector<pair<int,int>>vct[500000];
pair<pair<long long,long long>,map<long long,long long>>pr[500000];
long long ans=LONG_LONG_MAX,k;
void dfs(int in){
    if(vct[in].empty()){
        pr[in].second[0]=0;
        return;
    }
    for(auto &i:vct[in])dfs(i.first);
    int f=0,g=0;
    for(int i=0;i<vct[in].size();i++){
        if(pr[vct[in][i].first].second.size()>f){
            f=pr[vct[in][i].first].second.size();
            g=i;
        }
    }
    swap(pr[in],pr[vct[in][g].first]);
    pr[in].first.first+=vct[in][g].second;
    pr[in].first.second++;
    // cout<<'g'<<in<<' '<<pr[in].first.first<<' '<<pr[in].first.second<<endl;
    pr[in].second[-pr[in].first.first]=-pr[in].first.second;
    if(pr[in].second.count(k-pr[in].first.first)){
        ans=min(ans,(long long)pr[in].second[k-pr[in].first.first]+pr[in].first.second);
        // cout<<'f'<<in<<' '<<ans<<endl;
    }
    for(int i=0;i<vct[in].size();i++){
        if(i==g)continue;
        for(auto &w:pr[vct[in][i].first].second){
            long long a=w.first+pr[vct[in][i].first].first.first+vct[in][i].second,b=w.second+pr[vct[in][i].first].first.second+1;
            long long G=k-a-pr[in].first.first;
            if(pr[in].second.count(G)){
                // cout<<in<<endl;
                ans=min(ans,(long long)pr[in].second[G]+pr[in].first.second+b);
            }
            a-=pr[in].first.first,b-=pr[in].first.second;
            if(!pr[in].second.count(a))pr[in].second[a]=b;
            else pr[in].second[a]=min(pr[in].second[a],b);
        }
        pr[vct[in][i].first].second.clear();
    }
}
int best_path(int n, int K, int h[][2], int l[]){
    k=K;
    vector<pair<int,int>>vec[n];
    for(int i=0;i<n-1;i++){
        vec[h[i][0]].push_back({h[i][1],l[i]});
        vec[h[i][1]].push_back({h[i][0],l[i]});
    }
    deque<int>dq={0};
    vector<bool>vis(n);
    vis[0]=1;
    while(dq.size()){
        int a=dq.front();
        dq.pop_front();
        for(auto &i:vec[a]){
            if(!vis[i.first]){
                vis[i.first]=1;
                vct[a].push_back(i);
                dq.push_back(i.first);
            }
        }
    }
    dfs(0);
    return (ans==LONG_LONG_MAX?-1:ans);
}

Compilation message (stderr)

race.cpp: In function 'void dfs(int)':
race.cpp:14:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |     for(int i=0;i<vct[in].size();i++){
      |                 ~^~~~~~~~~~~~~~~
race.cpp:15:46: warning: comparison of integer expressions of different signedness: 'std::map<long long int, long long int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   15 |         if(pr[vct[in][i].first].second.size()>f){
      |            ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~
race.cpp:29:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   29 |     for(int i=0;i<vct[in].size();i++){
      |                 ~^~~~~~~~~~~~~~~

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