Submission #994446

#	Time	Username	Problem	Language	Result	Execution time	Memory
994446		pannenkoek	Race (IOI11_race)	C++14	43 / 100	3043 ms	107604 KiB

race

#include <bits/stdc++.h>
using namespace std;

#define rep(i, a, b) for(int i = (a); i < (b); i++)
#define pb push_back
#define fi first
#define se second

using ll = long long;
using ld = long double;
using pii = pair<ll, ll>;

const int MAXN = 2e5 + 5;
vector<pii> adj[MAXN];
map<ll, ll> poss[MAXN]; 
ll off[MAXN];
ll res = 1e18;
int k;

void dfs(ll v, ll p){
    poss[v][0] = 0;
    for(auto [u, d]: adj[v]){
        if(u == p) continue;
        dfs(u, v);
        // off[u] += d;
        // if(poss[u].size() > poss[v].size()) {
            // swap(off[u], off[v]);
            // swap(poss[u], poss[v]);
        // }
        set<ll> new_entries;
        map<ll, ll> old_vals;
        for(auto [dis, cost]: poss[u]){
            ll real_dis = dis + d;
            if(real_dis > k) continue;
            if(poss[v].count(k - real_dis) && !new_entries.count(k - real_dis)){
                ll old_cost = poss[v][k - real_dis];
                if(old_vals.count(k - real_dis)) {
                    old_cost = old_vals[k - real_dis]; 
                }
                res = min(old_cost + 1 + cost, res);
            }
            if(poss[v].count(real_dis)) {
                old_vals[real_dis] = poss[v][real_dis];
                poss[v][real_dis] = min(poss[v][real_dis], cost + 1);
            } else {
                poss[v][real_dis] = cost + 1;
                new_entries.insert(real_dis);
            }
        }
        poss[u] = {};
    }
}

int best_path(int N, int K, int H[][2], int L[]) {
    rep(i, 0, N) {
        adj[i] = {};
        poss[i] = {};
        off[i] = 0;
    }
    k = K;
    rep(i, 0, N - 1) {
        adj[H[i][0]].pb({H[i][1], L[i]});
        adj[H[i][1]].pb({H[i][0], L[i]});
    }
    dfs(0, -1); 
    if(res < 1e18) return res;
    else return -1;
}

Compilation message (stderr)

race.cpp: In function 'void dfs(ll, ll)':
race.cpp:22:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   22 |     for(auto [u, d]: adj[v]){
      |              ^
race.cpp:32:18: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   32 |         for(auto [dis, cost]: poss[u]){
      |                  ^

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