Submission #433979

#	Time	Username	Problem	Language	Result	Execution time	Memory
433979		illyakr	Race (IOI11_race)	C++14	43 / 100	3077 ms	41652 KiB

race

#include <iostream>
#include <vector>
#include <unordered_map>
#include "race.h"
using namespace std;

vector<pair<int, int> > g[201010];
int sz[201010];
bool del[201010];
void rcn(int v, int pred) {
    sz[v] = 1;
    for (auto [to, c] : g[v]) {
        if (to == pred || del[to]) continue;
        rcn(to, v);
        sz[v] += sz[to];
    }
}
int centroid(int v, int pred, int N) {
    for (auto [to, c] : g[v]) {
        if (to == pred || del[to])continue;
        if (sz[to] > N / 2)
            return centroid(to, v, N);
    }
    return v;
}
int Kur;
unordered_map<int, int> have;
int res = 1010101010;
void dfs(int v, int pred, int sum, int depth) {
    if (have.count(Kur - sum) != 0)
        res = min(res, have[Kur - sum] + depth);
    for (auto [to, c] : g[v]) {
        if (to == pred || del[to] || sum + c > Kur)continue;
        dfs(to, v, sum + c, depth + 1);
    }
}
void put(int v, int pred, int sum, int depth) {
    if (have.count(sum) != 0)
        have[sum] = min(have[sum], depth);
    else
        have[sum] = depth;
    for (auto [to, c] : g[v]) {
        if (to == pred || del[to] || sum + c > Kur)continue;
        put(to, v, sum + c, depth + 1);
    }
}

void decomposition(int v) {
    del[v] = true;
    rcn(v, v);
    have[0] = 0;
    for (auto [to, c] : g[v]) {
        if (del[to] || c > Kur)continue;
        dfs(to, v, c, 1);
        put(to, v, c, 1);
    }
    for (auto [to, c] : g[v]) {
        if (del[to])continue;
        have.clear();
        decomposition(centroid(to, v, sz[to]));
    }

}

int best_path(int N, int K, int H[][2], int L[]) {
    for (int i = 0; i < N - 1; i++) {
        g[H[i][0] + 1].push_back({H[i][1] + 1, L[i]});
        g[H[i][1] + 1].push_back({H[i][0] + 1, L[i]});
    }
    Kur = K;
    rcn(1, 1);
    decomposition(centroid(1, 1, N));
//    cout << centroid(1, 1, N) << " !!" << endl;

    if (res == 1010101010)return -1;
    return res;
}

/**
3 3
0 1 1
1 2 1
-1

11 12
0 1 3
0 2 4
2 3 5
3 4 4
4 5 6
0 6 3
6 7 2
6 8 5
8 9 6
8 10 7
2
*/

Compilation message (stderr)

race.cpp: In function 'void rcn(int, int)':
race.cpp:12:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   12 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp: In function 'int centroid(int, int, int)':
race.cpp:19:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   19 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp: In function 'void dfs(int, int, int, int)':
race.cpp:32:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   32 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp: In function 'void put(int, int, int, int)':
race.cpp:42:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   42 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp: In function 'void decomposition(int)':
race.cpp:52:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   52 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp:57:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   57 |     for (auto [to, c] : g[v]) {
      |               ^

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