Submission #433929

#	Time	Username	Problem	Language	Result	Execution time	Memory
433929		illyakr	Race (IOI11_race)	C++14	0 / 100	3 ms	4940 KiB

race

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

vector<pair<int, int> > g[201010];
int sz[201010];
bool del[201010];
void rcn(int v, int pred) {
    if (del[v])return;
    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) {
    if (del[v])return -1;
    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;
map<int, int> have;
int res = 1010101010;
void dfs(int v, int pred, int sum, int depth) {
    if (del[v])return;
    if (sum > Kur)return;
    if (sum == Kur)res = min(res, depth);
//    cout << v << " <<<< !!!!!!   " << sum << " " << depth << endl;
    if (have.count(Kur - sum) != 0) {
//        cout << "FIND" << endl;
        res = min(res, have[Kur - sum] + depth);
    } else {
//        cout << "How??  " << Kur - sum << endl;
    }
    for (auto [to, c] : g[v]) {
        if (to == pred || del[to])continue;
        dfs(to, v, sum + c, depth + 1);
    }
}
void put(int v, int pred, int sum, int depth) {
    if (del[v])return;
    if (sum > Kur)return;
    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])continue;
        dfs(to, v, sum + c, depth + 1);
    }
}

void decomposition(int v) {
    del[v] = true;
    rcn(v, v);
    have.clear();
//    cout << v << " !!!" << endl;
    for (auto [to, c] : g[v]) {
        if (del[to])continue;
//        cout << to << "  <<< " << endl;
        dfs(to, to, c, 1);
        put(to, to, c, 1);
//        for (auto [i, j] : have)cout << i << " " << j << "  ---- " << endl;
    }

    for (auto [to, c] : g[v]) {
        if (del[to])continue;
        decomposition(centroid(to, to, 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:11:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   11 |     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:40:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   40 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp: In function 'void put(int, int, int, 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: In function 'void decomposition(int)':
race.cpp:63:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   63 |     for (auto [to, c] : g[v]) {
      |               ^
race.cpp:71:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   71 |     for (auto [to, c] : g[v]) {
      |               ^

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