Submission #1037290

# Submission time Handle Problem Language Result Execution time Memory
1037290 2024-07-28T12:57:43 Z yanb Shortcut (IOI16_shortcut) C++14
Compilation error
0 ms 0 KB
#include <bits/stdc++.h>
    
using namespace std;

#define int long long
#define pii pair<long long, long long>

int dfs(int n, vector<vector<pii>> &g, int v, int p) {
    int ans = 0;
    for (auto [u, du] : g[v]) {
        if (u != p) {
            ans = max(ans, dfs(u, g, u, v) + du);
        }
    }
    return ans;
}

int diam(int n, vector<int> l, vector<int> d) {
    vector<vector<pii>> g(2 * n);
    for (int i = 0; i < n - 1; i++) {
        g[i].push_back({i + 1, l[i]});
        g[i + 1].push_back({i, l[i]});
    }
    for (int i = 0; i < n; i++) {
        g[i].push_back({n + i, d[i]});
        g[n + i].push_back({i, d[i]});
    }

    int ans = 0;
    for (int i = 0; i < 2 * n; i++) ans = max(ans, dfs(n, g, i, i));
    return ans;
}

int find_shortcut(int n, vector<int> l, vector<int> d, int c) {
    int noshort = diam(n, l, d);

    int ll = 0, rr = 1e17;
    while (rr - ll > 1) {
        int mm = (ll + rr) / 2;
        int Xmn = -1e17, Ymn = -1e17, Xmx = 1e17, Ymx = 1e17; // X = y + z, Y = y - z  
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n; j++) {
                int goal = mm - c - d[i] - d[j];
                int zavg = l[j], ymin = l[i] - goal, ymax = l[i] + goal;
                Xmn = max(Xmn, ymin + zavg);
                Xmx = min(Xmx, ymax + zavg);
                Ymn = max(Ymn, ymin - zavg);
                Ymx = min(Ymx, ymax - zavg);
            }
        }

        bool ok = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                int X = l[i] + l[j], Y = l[i] - l[j];
                ok = ok || (Xmn <= X && X <= Xmx && Ymn <= Y && Y <= Ymx);
            }
        }

        if (ok) rr = mm;
        else ll = mm;
    }
    return min(rr, noshort);
}

#ifdef LOCAL
signed main() {
    cout << find_shortcut(4, {10, 20, 20}, {0, 40, 0, 30}, 10) << "\n";
    cout << find_shortcut(9, {10, 10, 10, 10, 10, 10, 10, 10}, {20, 0, 30, 0, 0, 40, 0, 40, 0}, 30) << "\n";
    cout << find_shortcut(4, {2, 2, 2}, {1, 10, 10, 1}, 1) << "\n";
    cout << find_shortcut(3, {1, 1}, {1, 1, 1}, 3) << "\n";
    cout << find_shortcut(2, {10}, {1, 1}, 1) << "\n";
}
#endif

Compilation message

shortcut.cpp: In function 'long long int dfs(long long int, std::vector<std::vector<std::pair<long long int, long long int> > >&, long long int, long long int)':
shortcut.cpp:10:15: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   10 |     for (auto [u, du] : g[v]) {
      |               ^
/usr/bin/ld: /tmp/cckHfSKC.o: in function `main':
grader.cpp:(.text.startup+0x124): undefined reference to `find_shortcut(int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >, int)'
collect2: error: ld returned 1 exit status