# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
1037290 |
2024-07-28T12:57:43 Z |
yanb |
Shortcut (IOI16_shortcut) |
C++14 |
|
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